Skip to content

Commit 17ced2a

Browse files
committed
Improve default GUI values
1 parent 4a81eac commit 17ced2a

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

app.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,13 +19,13 @@ def browse_dataset_path():
1919

2020
# Initialize commonly used session state keys
2121
st.session_state.setdefault("dataset_path", "")
22-
st.session_state.setdefault("dataset_type", "COCO")
23-
st.session_state.setdefault("split", "val")
22+
st.session_state.setdefault("dataset_type", "YOLO")
23+
st.session_state.setdefault("split", "test")
2424
st.session_state.setdefault("config_option", "Manual Configuration")
2525
st.session_state.setdefault("confidence_threshold", 0.5)
2626
st.session_state.setdefault("nms_threshold", 0.5)
2727
st.session_state.setdefault("max_detections", 100)
28-
st.session_state.setdefault("device", "cpu")
28+
st.session_state.setdefault("device", "cuda")
2929
st.session_state.setdefault("batch_size", 1)
3030
st.session_state.setdefault("evaluation_step", 5)
3131
st.session_state.setdefault("detection_model", None)

0 commit comments

Comments
 (0)