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

View File

@ -71,6 +71,7 @@ def train():
train_path = data_dict['train']
test_path = data_dict['valid']
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
for f in glob.glob('*_batch*.png') + glob.glob(results_file):