fix config and det

paddlex/inference/components/transforms/image/common.py

@@ -37,6 +37,7 @@ __all__ = [
     "Pad",
     "Normalize",
     "ToCHWImage",
+    "PadStride",
 ]
 
 
@@ -486,6 +487,7 @@ class PadStride(BaseComponent):
     DEAULT_OUTPUTS = {"img": "img"}
 
     def __init__(self, stride=0):
+        super().__init__()
         self.coarsest_stride = stride
 
     def apply(self, img):

paddlex/inference/models/object_detection.py

@@ -44,7 +44,7 @@ class DetPredictor(CVPredictor):
             model_prefix=self.MODEL_FILE_PREFIX,
             option=self.pp_option,
         )
-        if "DETR" in self.model_name:
+        if "DETR" in self.model_name or "RCNN" in self.model_name:
             predictor.set_inputs(
                 {
                     "img": "img",

paddlex/inference/utils/official_models.py

@@ -156,10 +156,10 @@ CLIP_vit_large_patch14_224_infer.tar",
 Deeplabv3_Plus-R50_infer.tar",
     "Deeplabv3_Plus-R101": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0b1_v2/\
 Deeplabv3_Plus-R101_infer.tar",
-    "PP-ShiTuV2_rec": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0/PP-ShiTuV2_rec_infer.tar",
-    "PP-ShiTuV2_rec_CLIP_vit_base": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0/\
+    "PP-ShiTuV2_rec": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0b1_v2/PP-ShiTuV2_rec_infer.tar",
+    "PP-ShiTuV2_rec_CLIP_vit_base": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0b1_v2/\
 PP-ShiTuV2_rec_CLIP_vit_base_infer.tar",
-    "PP-ShiTuV2_rec_CLIP_vit_large": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0/\
+    "PP-ShiTuV2_rec_CLIP_vit_large": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0b1_v2/\
 PP-ShiTuV2_rec_CLIP_vit_large_infer.tar",
     "PP-LiteSeg-T": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0b1_v2/PP-LiteSeg-T_infer.tar",
     "OCRNet_HRNet-W48": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0b1_v2/OCRNet_HRNet-W48_infer.tar",