feat 环境更新
This commit is contained in:
@@ -30,9 +30,6 @@ class AttributeRecognition:
|
||||
self.const = const
|
||||
self.triton_client = httpclient.InferenceServerClient(url=f"{ATT_TRITON_URL}")
|
||||
|
||||
def __del__(self):
|
||||
self.triton_client.close()
|
||||
|
||||
def get_result(self):
|
||||
for sketch in self.request_data:
|
||||
if sketch['category'] == "Tops" or sketch['category'] == "Blouse":
|
||||
|
||||
@@ -64,11 +64,6 @@ class GenerateImage:
|
||||
|
||||
pass
|
||||
|
||||
def __del__(self):
|
||||
self.redis_client.close()
|
||||
self.triton_client.close()
|
||||
self.connection.close()
|
||||
|
||||
@staticmethod
|
||||
def image_grid(imgs, rows, cols):
|
||||
assert len(imgs) == rows * cols
|
||||
|
||||
@@ -29,11 +29,6 @@ class SuperResolution:
|
||||
self.connection = pika.BlockingConnection(pika.ConnectionParameters(**RABBITMQ_PARAMS))
|
||||
self.channel = self.connection.channel()
|
||||
|
||||
def __del__(self):
|
||||
self.redis_client.close()
|
||||
self.triton_client.close()
|
||||
self.connection.close()
|
||||
|
||||
# @RunTime
|
||||
def read_image(self):
|
||||
try:
|
||||
|
||||
Reference in New Issue
Block a user