change imports

supervisely-ecosystem · Jul 25, 2024 · 883d39b · 883d39b
1 parent 21d8728
commit 883d39b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,2 +1,2 @@
-git+https://github.com/supervisely/supervisely.git@model-benchmark-ars-before-merge-with-main-branch
-ujson
+ujson
+git+https://github.com/supervisely/supervisely.git@model-benchmark-ars-before-merge-with-main-branch
diff --git a/src/main.py b/src/main.py
@@ -10,12 +10,13 @@
 # from src.ui.outcome_counts import plotly_outcome_counts
 from supervisely._utils import camel_to_snake
 from supervisely.app.widgets import *
+from supervisely.nn.benchmark.object_detection_benchmark import ObjectDetectionBenchmark
 
 
 def main_func(api: sly.Api):
     project = api.project.get_info_by_id(g.project_id)
 
-    bm = sly.nn.ObjectDetectionBenchmark(api, project.id, output_dir=g.STORAGE_DIR + "/benchmark")
+    bm = ObjectDetectionBenchmark(api, project.id, output_dir=g.STORAGE_DIR + "/benchmark")
     sly.logger.info("Session ID={}".format(g.session_id))
     bm.run_evaluation(model_session=g.session_id)
     # bm.evaluate(g.dt_project_id)