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 6fe485f2ab
commit 13e3f8ac3d
4 changed files with 5 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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")