From c7f93bae403ed9cf9bd50319a29485643d2438ad Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 6 Apr 2020 10:58:07 -0700 Subject: [PATCH] update coco-tuned hyp['cls'] to current dataset --- train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/train.py b/train.py index 802997ac..9198f623 100644 --- a/train.py +++ b/train.py @@ -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):