Update train.py (#5014)
Browse files
train.py
CHANGED
@@ -412,7 +412,7 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary
|
|
412 |
batch_size=batch_size // WORLD_SIZE * 2,
|
413 |
imgsz=imgsz,
|
414 |
model=attempt_load(f, device).half(),
|
415 |
-
iou_thres=0.
|
416 |
single_cls=single_cls,
|
417 |
dataloader=val_loader,
|
418 |
save_dir=save_dir,
|
|
|
412 |
batch_size=batch_size // WORLD_SIZE * 2,
|
413 |
imgsz=imgsz,
|
414 |
model=attempt_load(f, device).half(),
|
415 |
+
iou_thres=0.65 if is_coco else 0.60, # best pycocotools results at 0.65
|
416 |
single_cls=single_cls,
|
417 |
dataloader=val_loader,
|
418 |
save_dir=save_dir,
|