zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

--rect test bug fix

This commit is contained in:
Glenn Jocher 2020-07-09 20:39:11 -07:00
parent d40e4781f1
commit 6649770001
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test.py
View File

@ -59,7 +59,7 @@ def test(data,
_ = model(img.half() if half else img) if device.type != 'cpu' else None # run once _ = model(img.half() if half else img) if device.type != 'cpu' else None # run once
path = data['test'] if opt.task == 'test' else data['val'] # path to val/test images path = data['test'] if opt.task == 'test' else data['val'] # path to val/test images
dataloader = create_dataloader(path, imgsz, batch_size, model.stride.max(), opt, dataloader = create_dataloader(path, imgsz, batch_size, model.stride.max(), opt,
hyp=None, augment=False, cache=False, pad=0.5, rect=False)[0] hyp=None, augment=False, cache=False, pad=0.5, rect=True)[0]
seen = 0 seen = 0
names = model.names if hasattr(model, 'names') else model.module.names names = model.names if hasattr(model, 'names') else model.module.names

1
utils/datasets.py
View File

@ -343,6 +343,7 @@ class LoadImagesAndLabels(Dataset): # for training/testing
irect = ar.argsort() irect = ar.argsort()
self.img_files = [self.img_files[i] for i in irect] self.img_files = [self.img_files[i] for i in irect]
self.label_files = [self.label_files[i] for i in irect] self.label_files = [self.label_files[i] for i in irect]
self.labels = [self.labels[i] for i in irect]
self.shapes = s[irect] # wh self.shapes = s[irect] # wh
ar = ar[irect] ar = ar[irect]