diff --git a/test.py b/test.py index 5536eab0..222f829a 100644 --- a/test.py +++ b/test.py @@ -15,7 +15,7 @@ def test( batch_size=16, img_size=416, iou_thres=0.5, - conf_thres=0.1, + conf_thres=0.001, nms_thres=0.5, save_json=False, model=None diff --git a/train.py b/train.py index ba134bc4..94e5e1e7 100644 --- a/train.py +++ b/train.py @@ -180,7 +180,7 @@ def train( # Calculate mAP with torch.no_grad(): - results = test.test(cfg, data_cfg, batch_size=batch_size, img_size=img_size, model=model) + results = test.test(cfg, data_cfg, batch_size=batch_size, img_size=img_size, model=model, conf_thres=0.1) # Write epoch results with open('results.txt', 'a') as file: