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

feat design相关模型迁移4090测试

Browse Source 
fix
This commit is contained in:
zhouchengrong
2024-11-08 14:05:09 +08:00
parent 0a17128140
commit d92c59383b
4 changed files with 5 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -28,7 +28,7 @@ class AttributeRecognition:
}
)
self.const = const
self.triton_client = httpclient.InferenceServerClient(url=f"{ATT_TRITON_URL}")
self.triton_client = httpclient.InferenceServerClient(url=f"{DESIGN_MODEL_URL}")
def get_result(self):
for sketch in self.request_data:

2
app/service/attribute/service_category_recognition.py
View File

@@ -26,7 +26,7 @@ class CategoryRecognition:
self.attr_type = pd.read_csv(CATEGORY_PATH)
# self.minio_client = Minio(MINIO_URL, access_key=MINIO_ACCESS, secret_key=MINIO_SECRET, secure=MINIO_SECURE)
self.request_data = []
self.triton_client = httpclient.InferenceServerClient(url=ATT_TRITON_URL)
self.triton_client = httpclient.InferenceServerClient(url=DESIGN_MODEL_URL)
for sketch in request_data:
self.request_data.append(
{

4
app/service/generate_image/utils/image_processing.py
View File

@@ -81,7 +81,7 @@ def get_contours(image):
def seg_infer_image(image_obj):
image, ori_shape = seg_preprocess(image_obj)
client = httpclient.InferenceServerClient(url=f"{SEG_MODEL_URL}")
client = httpclient.InferenceServerClient(url=f"{DESIGN_MODEL_URL}")
transformed_img = image.astype(np.float32)
# 输入集
inputs = [
@@ -250,7 +250,7 @@ def generate_category_recognition(image, gender):
return preprocessed_img
preprocessed_img = preprocess(image)
triton_client = httpclient.InferenceServerClient(url=ATT_TRITON_URL)
triton_client = httpclient.InferenceServerClient(url=DESIGN_MODEL_URL)
inputs = [
httpclient.InferInput("input__0", preprocessed_img.shape, datatype="FP32")