Skip to content

Commit

Permalink
updated progress
Browse files Browse the repository at this point in the history
  • Loading branch information
NikolaiPetukhov committed Sep 13, 2024
1 parent 0061422 commit be9f398
Show file tree
Hide file tree
Showing 2 changed files with 9 additions and 1 deletion.
1 change: 1 addition & 0 deletions requirements.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
git+https://github.com/supervisely/supervisely.git@speedup-inference#egg=supervisely[model-benchmark]
9 changes: 8 additions & 1 deletion src/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -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":
Expand All @@ -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 = }")
Expand Down Expand Up @@ -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)
Expand Down Expand Up @@ -108,14 +112,17 @@ def main_func():
button.disable()

pbar = widgets.SlyTqdm()
sec_pbar = widgets.Progress("")

report_model_benchmark = widgets.ReportThumbnail()
report_model_benchmark.hide()

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]
),
)


Expand Down

0 comments on commit be9f398

Please sign in to comment.