Spaces:

fcakyon
/

sahi-yolox

Running

fcakyon commited on Nov 13, 2022

Commit

fe27809

1 Parent(s): a7ccadc

update to latest sahi api

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import streamlit as st
 import sahi.utils.file
 import sahi.utils.mmdet
-import sahi.model
 from PIL import Image
 import random
 from utils import sahi_mmdet_inference
@@ -42,7 +42,8 @@ def get_model():
         model_name="yolox", config_file_name="yolox_tiny_8x8_300e_coco.py"
     )
-    detection_model = sahi.model.MmdetDetectionModel(
         model_path=model_path,
         config_path=config_path,
         confidence_threshold=0.5,

 import streamlit as st
 import sahi.utils.file
 import sahi.utils.mmdet
+from sahi import AutoDetectionModel
 from PIL import Image
 import random
 from utils import sahi_mmdet_inference
         model_name="yolox", config_file_name="yolox_tiny_8x8_300e_coco.py"
     )
+    detection_model = AutoDetectionModel(
+        model_type='mmdet'
         model_path=model_path,
         config_path=config_path,
         confidence_threshold=0.5,