Skip to content

Commit

Permalink
add comment and remove no need param
Browse files Browse the repository at this point in the history
  • Loading branch information
shensheng272 authored Sep 20, 2022
1 parent ecf0ccc commit 9a1bb03
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions configs/experiment/yolov6n_with_eval_params.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,11 +54,14 @@
mixup=0.0,
)

# Eval params when eval model.
# If eval_params item is list, eg conf_thres=[0.03, 0.03],
# first will be used in train.py and second will be used in eval.py.
eval_params = dict(
batch_size=None, #None mean will be the same as batch on one device * 2
img_size=None, #None mean will be the same as train image size
conf_thres=0.03,
iou_thres=0.65,
task='train',

#pading and scale coord
test_load_size=None, #None mean will be the same as test image size
Expand All @@ -73,4 +76,4 @@
do_pr_metric=False,
plot_curve=False,
plot_confusion_matrix=False
)
)

0 comments on commit 9a1bb03

Please sign in to comment.