Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

update coco-tuned hyp['cls'] to current dataset

This commit is contained in:
Glenn Jocher 2020-04-06 10:58:07 -07:00
parent 4fc0012829
commit c7f93bae40
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
train.py
View File

@ -71,6 +71,7 @@ def train():
train_path = data_dict['train'] train_path = data_dict['train']
test_path = data_dict['valid'] test_path = data_dict['valid']
nc = 1 if opt.single_cls else int(data_dict['classes']) # number of classes nc = 1 if opt.single_cls else int(data_dict['classes']) # number of classes
hyp['cls'] *= nc / 80 # update coco-tuned hyp['cls'] to current dataset
# Remove previous results # Remove previous results
for f in glob.glob('*_batch*.png') + glob.glob(results_file): for f in glob.glob('*_batch*.png') + glob.glob(results_file):