From 6ab0a366f47d8470f48ddea3911c5513acd936b0 Mon Sep 17 00:00:00 2001 From: iumyx2612 <69593462+iumyx2612@users.noreply.github.com> Date: Thu, 2 Dec 2021 22:11:44 +0700 Subject: [PATCH 1/3] Update val.py Add an option to choose number of workers if not called by train.py --- val.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/val.py b/val.py index 27edd158a2f6..391c89d340dc 100644 --- a/val.py +++ b/val.py @@ -89,6 +89,7 @@ def run(data, iou_thres=0.6, # NMS IoU threshold task='val', # train, val, test, speed or study device='', # cuda device, i.e. 0 or 0,1,2,3 or cpu + workers=8, single_cls=False, # treat as single-class dataset augment=False, # augmented inference verbose=False, # verbose output @@ -152,7 +153,7 @@ def run(data, model.warmup(imgsz=(1, 3, imgsz, imgsz), half=half) # warmup pad = 0.0 if task == 'speed' else 0.5 task = task if task in ('train', 'val', 'test') else 'val' # path to train/val/test images - dataloader = create_dataloader(data[task], imgsz, batch_size, stride, single_cls, pad=pad, rect=pt, + dataloader = create_dataloader(data[task], imgsz, batch_size, stride, single_cls, pad=pad, rect=pt, workers=workers, prefix=colorstr(f'{task}: '))[0] seen = 0 @@ -312,6 +313,7 @@ def parse_opt(): parser.add_argument('--iou-thres', type=float, default=0.6, help='NMS IoU threshold') parser.add_argument('--task', default='val', help='train, val, test, speed or study') parser.add_argument('--device', default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu') + parser.add_argument('--workers', type=int, default=8, help='max dataloader workers (per RANK in DDP mode)') parser.add_argument('--single-cls', action='store_true', help='treat as single-class dataset') parser.add_argument('--augment', action='store_true', help='augmented inference') parser.add_argument('--verbose', action='store_true', help='report mAP by class') From 4cb928ce60873b93e7e5a72adeb1791d4e4ed0a0 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 2 Dec 2021 16:35:05 +0100 Subject: [PATCH 2/3] Update comment --- val.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/val.py b/val.py index 391c89d340dc..eb60cdcdfbab 100644 --- a/val.py +++ b/val.py @@ -89,7 +89,7 @@ def run(data, iou_thres=0.6, # NMS IoU threshold task='val', # train, val, test, speed or study device='', # cuda device, i.e. 0 or 0,1,2,3 or cpu - workers=8, + workers=8, # max dataloader workers (per RANK in DDP mode) single_cls=False, # treat as single-class dataset augment=False, # augmented inference verbose=False, # verbose output From 41a068d92257f05a5b54041057e4158603d738f7 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 2 Dec 2021 16:36:29 +0100 Subject: [PATCH 3/3] 120 char line width --- val.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/val.py b/val.py index eb60cdcdfbab..4eec499d3029 100644 --- a/val.py +++ b/val.py @@ -153,8 +153,8 @@ def run(data, model.warmup(imgsz=(1, 3, imgsz, imgsz), half=half) # warmup pad = 0.0 if task == 'speed' else 0.5 task = task if task in ('train', 'val', 'test') else 'val' # path to train/val/test images - dataloader = create_dataloader(data[task], imgsz, batch_size, stride, single_cls, pad=pad, rect=pt, workers=workers, - prefix=colorstr(f'{task}: '))[0] + dataloader = create_dataloader(data[task], imgsz, batch_size, stride, single_cls, pad=pad, rect=pt, + workers=workers, prefix=colorstr(f'{task}: '))[0] seen = 0 confusion_matrix = ConfusionMatrix(nc=nc)