Procházet zdrojové kódy

delete input_shpae

syyxsxx před 5 roky
rodič
revize
dce71e19ab
1 změnil soubory, kde provedl 3 přidání a 0 odebrání
  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":