瀏覽代碼

fix instance_seg bug

zhangyubo0722 1 年之前
父節點
當前提交
5ad595ae45
共有 2 個文件被更改,包括 3 次插入3 次删除
  1. 1 1
      paddlex/inference/models/instance_segmentation.py
  2. 2 2
      paddlex/inference/utils/official_models.py

+ 1 - 1
paddlex/inference/models/instance_segmentation.py

@@ -40,7 +40,7 @@ class InstanceSegPredictor(DetPredictor):
             model_prefix=self.MODEL_FILE_PREFIX,
             option=self.pp_option,
         )
-        model_names = ["RT-DETR", "SOLOv2", "RCNN"]
+        model_names = ["RT-DETR", "SOLOv2", "RCNN", "YOLO"]
         if any(name in self.model_name for name in model_names):
             predictor.set_inputs(
                 {"img": "img", "scale_factors": "scale_factors", "img_size": "img_size"}

+ 2 - 2
paddlex/inference/utils/official_models.py

@@ -203,9 +203,9 @@ PP-OCRv4_server_seal_det_infer.tar",
     "PP-OCRv4_mobile_seal_det": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0b1_v2/\
 PP-OCRv4_mobile_seal_det_infer.tar",
     "ch_RepSVTR_rec": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0b1_v2/\
-openatom_rec_repsvtr_ch_infer.tar",
+ch_RepSVTR_rec_infer.tar",
     "ch_SVTRv2_rec": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0b1_v2/\
-openatom_rec_svtrv2_ch_infer.tar",
+ch_SVTRv2_rec_infer.tar",
     "PP-LCNet_x1_0_pedestrian_attribute": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0b1_v2/\
 PP-LCNet_x1_0_pedestrian_attribute_infer.tar",
     "PP-LCNet_x1_0_vehicle_attribute": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0b1_v2/\