xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 13:34:23 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Adds toggle displaying labels in GUI and verbose log on start (#7804)

Browse Source 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Ayush Chaurasia 2024-01-25 22:05:46 +05:30 committed by GitHub
parent c460ad28f8
commit 62742c21d2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 24 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ultralytics/cfg/__init__.py
View File

@ -396,6 +396,7 @@ def handle_yolo_settings(args: List[str]) -> None:
def handle_explorer(): def handle_explorer():
"""Open the Ultralytics Explorer GUI.""" """Open the Ultralytics Explorer GUI."""
checks.check_requirements("streamlit") checks.check_requirements("streamlit")
LOGGER.info(f"💡 Loading Explorer dashboard...")
subprocess.run(["streamlit", "run", ROOT / "data/explorer/gui/dash.py", "--server.maxMessageSize", "2048"]) subprocess.run(["streamlit", "run", ROOT / "data/explorer/gui/dash.py", "--server.maxMessageSize", "2048"])

26
ultralytics/data/explorer/gui/dash.py
View File

@ -9,7 +9,7 @@ from ultralytics import Explorer
from ultralytics.utils import ROOT, SETTINGS from ultralytics.utils import ROOT, SETTINGS
from ultralytics.utils.checks import check_requirements from ultralytics.utils.checks import check_requirements
check_requirements(("streamlit>=1.29.0", "streamlit-select>=0.2")) check_requirements(("streamlit>=1.29.0", "streamlit-select>=0.3"))
import streamlit as st import streamlit as st
from streamlit_select import image_select from streamlit_select import image_select
@ -94,6 +94,7 @@ def find_similar_imgs(imgs):
similar = exp.get_similar(img=imgs, limit=st.session_state.get("limit"), return_type="arrow") similar = exp.get_similar(img=imgs, limit=st.session_state.get("limit"), return_type="arrow")
paths = similar.to_pydict()["im_file"] paths = similar.to_pydict()["im_file"]
st.session_state["imgs"] = paths st.session_state["imgs"] = paths
st.session_state["res"] = similar
def similarity_form(selected_imgs): def similarity_form(selected_imgs):
@ -137,6 +138,7 @@ def run_sql_query():
exp = st.session_state["explorer"] exp = st.session_state["explorer"]
res = exp.sql_query(query, return_type="arrow") res = exp.sql_query(query, return_type="arrow")
st.session_state["imgs"] = res.to_pydict()["im_file"] st.session_state["imgs"] = res.to_pydict()["im_file"]
st.session_state["res"] = res
def run_ai_query(): def run_ai_query():
@ -155,6 +157,7 @@ def run_ai_query():
st.session_state["error"] = "No results found using AI generated query. Try another query or rerun it." st.session_state["error"] = "No results found using AI generated query. Try another query or rerun it."
return return
st.session_state["imgs"] = res["im_file"].to_list() st.session_state["imgs"] = res["im_file"].to_list()
st.session_state["res"] = res
def reset_explorer(): def reset_explorer():
@ -195,7 +198,11 @@ def layout():
if st.session_state.get("error"): if st.session_state.get("error"):
st.error(st.session_state["error"]) st.error(st.session_state["error"])
else: else:
imgs = st.session_state.get("imgs") or exp.table.to_lance().to_table(columns=["im_file"]).to_pydict()["im_file"] if st.session_state.get("imgs"):
imgs = st.session_state.get("imgs")
else:
imgs = exp.table.to_lance().to_table(columns=["im_file"]).to_pydict()["im_file"]
st.session_state["res"] = exp.table.to_arrow()
total_imgs, selected_imgs = len(imgs), [] total_imgs, selected_imgs = len(imgs), []
with col1: with col1:
subcol1, subcol2, subcol3, subcol4, subcol5 = st.columns(5) subcol1, subcol2, subcol3, subcol4, subcol5 = st.columns(5)
@ -230,17 +237,30 @@ def layout():
query_form() query_form()
ai_query_form() ai_query_form()
if total_imgs: if total_imgs:
labels, boxes, masks, kpts, classes = None, None, None, None, None
task = exp.model.task
if st.session_state.get("display_labels"):
labels = st.session_state.get("res").to_pydict()["labels"][start_idx : start_idx + num]
boxes = st.session_state.get("res").to_pydict()["bboxes"][start_idx : start_idx + num]
masks = st.session_state.get("res").to_pydict()["masks"][start_idx : start_idx + num]
kpts = st.session_state.get("res").to_pydict()["keypoints"][start_idx : start_idx + num]
classes = st.session_state.get("res").to_pydict()["cls"][start_idx : start_idx + num]
imgs_displayed = imgs[start_idx : start_idx + num] imgs_displayed = imgs[start_idx : start_idx + num]
selected_imgs = image_select( selected_imgs = image_select(
f"Total samples: {total_imgs}", f"Total samples: {total_imgs}",
images=imgs_displayed, images=imgs_displayed,
use_container_width=False, use_container_width=False,
# indices=[i for i in range(num)] if select_all else None, # indices=[i for i in range(num)] if select_all else None,
labels=labels,
classes=classes,
bboxes=boxes,
masks=masks if task == "segment" else None,
kpts=kpts if task == "pose" else None,
) )
with col2: with col2:
similarity_form(selected_imgs) similarity_form(selected_imgs)
# display_labels = st.checkbox("Labels", value=False, key="display_labels") display_labels = st.checkbox("Labels", value=False, key="display_labels")
utralytics_explorer_docs_callback() utralytics_explorer_docs_callback()