diff --git a/test.py b/test.py index b87288ee..e2709af7 100644 --- a/test.py +++ b/test.py @@ -7,7 +7,16 @@ from utils.utils import * from utils import torch_utils -def test(cfg, data_cfg, weights, batch_size=16, img_size=416, iou_thres=0.5, conf_thres=0.3, nms_thres=0.45): +def test( + cfg, + data_cfg, + weights, + batch_size=16, + img_size=416, + iou_thres=0.5, + conf_thres=0.3, + nms_thres=0.45 +): device = torch_utils.select_device() # Configure run @@ -125,5 +134,13 @@ if __name__ == '__main__': opt = parser.parse_args() print(opt, end='\n\n') - mAP = test(opt.cfg, opt.data_cfg, opt.weights, opt.batch_size, opt.img_size, opt.iou_thres, opt.conf_thres, - opt.nms_thres) + mAP = test( + opt.cfg, + opt.data_cfg, + opt.weights, + opt.batch_size, + opt.img_size, + opt.iou_thres, + opt.conf_thres, + opt.nms_thres + )