add fast model support for detector api

Signed-off-by: Bedapudi Praneeth <[email protected]>
WOLFBED · Mar 19, 2021 · af4d8e4 · af4d8e4
1 parent 6ebf237
commit af4d8e4
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/fastDeploy_recipes/detector/predictor.py b/fastDeploy_recipes/detector/predictor.py
@@ -20,7 +20,10 @@ def predictor(in_images=[], batch_size=32):
 
     for in_img in in_images:
         try:
-            preds.append(detector.detect(in_img))
+            if "_fast_mode_" in in_img:
+                preds.append(detector.detect(in_img, mode="fast"))
+            else:
+                preds.append(detector.detect(in_img))
         except:
             preds.append(None)