diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..f69bf1c --- /dev/null +++ b/requirements.txt @@ -0,0 +1 @@ +git+https://github.com/supervisely/supervisely.git@speedup-inference#egg=supervisely[model-benchmark] \ No newline at end of file diff --git a/src/main.py b/src/main.py index dc1fa6a..1f0ac4b 100644 --- a/src/main.py +++ b/src/main.py @@ -30,12 +30,14 @@ def main_func(): return pbar.show() + sec_pbar.show() if task_type == "object detection": bm = ObjectDetectionBenchmark( api, project.id, output_dir=g.STORAGE_DIR + "/benchmark", progress=pbar, + progress_secondary=sec_pbar, classes_whitelist=g.selected_classes, ) elif task_type == "instance segmentation": @@ -44,6 +46,7 @@ def main_func(): project.id, output_dir=g.STORAGE_DIR + "/benchmark", progress=pbar, + progress_secondary=sec_pbar, classes_whitelist=g.selected_classes, ) sly.logger.info(f"{g.session_id = }") @@ -75,6 +78,7 @@ def main_func(): report_model_benchmark.set(template_vis_file) report_model_benchmark.show() pbar.hide() + sec_pbar.hide() # ==================== Workflow output ==================== w.workflow_output(api, res_dir, template_vis_file) @@ -108,6 +112,7 @@ def main_func(): button.disable() pbar = widgets.SlyTqdm() +sec_pbar = widgets.Progress("") report_model_benchmark = widgets.ReportThumbnail() report_model_benchmark.hide() @@ -115,7 +120,9 @@ def main_func(): controls_card = widgets.Card( title="Settings", description="Select Ground Truth project and deployed model session", - content=widgets.Container([sel_project, sel_app_session, button, report_model_benchmark, pbar]), + content=widgets.Container( + [sel_project, sel_app_session, button, report_model_benchmark, pbar, sec_pbar] + ), )