From b36f4d0a887eb8b423d925478710669343b89b61 Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Mon, 24 Jun 2024 11:58:28 +0800 Subject: [PATCH] =?UTF-8?q?feat=20fix=20=20=E8=B6=85=E5=88=86=E5=BC=82?= =?UTF-8?q?=E5=B8=B8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/service/super_resolution/service.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/app/service/super_resolution/service.py b/app/service/super_resolution/service.py index f864d01..557beac 100644 --- a/app/service/super_resolution/service.py +++ b/app/service/super_resolution/service.py @@ -1,12 +1,14 @@ import json import logging import time + import cv2 import minio.error import numpy as np import redis import torch import tritonclient.grpc as grpcclient + from app.core.config import * from app.schemas.super_resolution import SuperResolutionModel from app.service.utils.oss_client import oss_get_image, oss_upload_image @@ -32,6 +34,7 @@ class SuperResolution: def read_image(self): try: img = oss_get_image(bucket=self.sr_image_url.split("/", 1)[0], object_name=self.sr_image_url.split("/", 1)[1], data_type="cv2") + img = img.astype(np.float32) / 255. # 解码 except minio.error.S3Error as e: sr_data = json.dumps({'tasks_id': self.tasks_id, 'status': 'ERROR', 'message': f'{e}'}) self.channel.basic_publish(exchange='', routing_key=SR_RABBITMQ_QUEUES, body=sr_data) @@ -144,6 +147,6 @@ def infer_cancel(tasks_id): if __name__ == '__main__': - request_data = SuperResolutionModel(sr_image_url="test/512_image/15.png", sr_xn=2, sr_tasks_id="123") + request_data = SuperResolutionModel(sr_image_url="test/512_image/15.png", sr_xn=2, sr_tasks_id="12341556") service = SuperResolution(request_data) result_url = service.sr_result()