Glenn Jocher
4 years ago
2 changed files with 2 additions and 1 deletions
+ 1
- 1
test.py
View File
| _ = 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
- 0
utils/datasets.py
View File
| 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] | ||||
Loading…