From 883d39bf54d85567d789b4a5ec5e3f2777c2b652 Mon Sep 17 00:00:00 2001 From: Ars Vladimirov Date: Thu, 25 Jul 2024 16:00:12 +0500 Subject: [PATCH] change imports --- requirements.txt | 4 ++-- src/main.py | 3 ++- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/requirements.txt b/requirements.txt index 0be0fee..7b8e4a2 100644 --- 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 \ No newline at end of file +ujson +git+https://github.com/supervisely/supervisely.git@model-benchmark-ars-before-merge-with-main-branch \ No newline at end of file diff --git a/src/main.py b/src/main.py index 219572a..f060c14 100644 --- 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)