瀏覽代碼

delete input_shpae

syyxsxx 5 年之前
父節點
當前提交
dce71e19ab
共有 1 個文件被更改,包括 3 次插入0 次删除
  1. 3 0
      deploy/openvino/python/deploy.py

+ 3 - 0
deploy/openvino/python/deploy.py

@@ -216,6 +216,9 @@ class Predictor:
         return result
 
     def predict(self, image, topk=1, threshold=0.5):
+        if self.model_type == "detector":
+            print("lite python api does not support detector")
+            return
         preprocessed_input = self.preprocess(image)
         model_pred = self.raw_predict(preprocessed_input)
         if self.model_type == "classifier":