aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat 环境更新

Browse Source
This commit is contained in:
zhouchengrong
2024-04-16 17:15:00 +08:00
parent 32ebb2232c
commit c9b407b2a4
3 changed files with 0 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
app/service/attribute/service_att_recognition.py
View File

@@ -30,9 +30,6 @@ class AttributeRecognition:
self.const = const self.const = const
self.triton_client = httpclient.InferenceServerClient(url=f"{ATT_TRITON_URL}") self.triton_client = httpclient.InferenceServerClient(url=f"{ATT_TRITON_URL}")
def __del__(self):
self.triton_client.close()
def get_result(self): def get_result(self):
for sketch in self.request_data: for sketch in self.request_data:
if sketch['category'] == "Tops" or sketch['category'] == "Blouse": if sketch['category'] == "Tops" or sketch['category'] == "Blouse":

5
app/service/generate_image/test.py
View File

@@ -64,11 +64,6 @@ class GenerateImage:
pass pass
def __del__(self):
self.redis_client.close()
self.triton_client.close()
self.connection.close()
@staticmethod @staticmethod
def image_grid(imgs, rows, cols): def image_grid(imgs, rows, cols):
assert len(imgs) == rows * cols assert len(imgs) == rows * cols

5
app/service/super_resolution/service.py
View File

@@ -29,11 +29,6 @@ class SuperResolution:
self.connection = pika.BlockingConnection(pika.ConnectionParameters(**RABBITMQ_PARAMS)) self.connection = pika.BlockingConnection(pika.ConnectionParameters(**RABBITMQ_PARAMS))
self.channel = self.connection.channel() self.channel = self.connection.channel()
def __del__(self):
self.redis_client.close()
self.triton_client.close()
self.connection.close()
# @RunTime # @RunTime
def read_image(self): def read_image(self):
try: try: