diff --git a/train.py b/train.py index 62a9bc62..f43489a4 100644 --- a/train.py +++ b/train.py @@ -301,7 +301,7 @@ def train(cfg, batch_size=batch_size, img_size=opt.img_size, model=model, - conf_thres=0.001 if final_epoch else 0.1, # 0.1 for speed + conf_thres=0.001 if final_epoch and epoch > 0 else 0.1, # 0.1 for speed save_json=final_epoch and 'coco.data' in data) # Write epoch results