File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -19,13 +19,13 @@ def browse_dataset_path():
1919
2020# Initialize commonly used session state keys
2121st .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 " )
2424st .session_state .setdefault ("config_option" , "Manual Configuration" )
2525st .session_state .setdefault ("confidence_threshold" , 0.5 )
2626st .session_state .setdefault ("nms_threshold" , 0.5 )
2727st .session_state .setdefault ("max_detections" , 100 )
28- st .session_state .setdefault ("device" , "cpu " )
28+ st .session_state .setdefault ("device" , "cuda " )
2929st .session_state .setdefault ("batch_size" , 1 )
3030st .session_state .setdefault ("evaluation_step" , 5 )
3131st .session_state .setdefault ("detection_model" , None )
You can’t perform that action at this time.
0 commit comments