Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2020-01-11 13:13:57 -08:00
parent 9b84885775
commit 4b56a370e6
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/datasets.py
View File

@ -499,7 +499,7 @@ class LoadImagesAndLabels(Dataset): # for training/testing
@staticmethod @staticmethod
def collate_fn(batch): def collate_fn(batch):
img, label, path, shapes = list(zip(*batch)) # transposed img, label, path, shapes = zip(*batch) # transposed
for i, l in enumerate(label): for i, l in enumerate(label):
l[:, 0] = i # add target image index for build_targets() l[:, 0] = i # add target image index for build_targets()
return torch.stack(img, 0), torch.cat(label, 0), path, shapes return torch.stack(img, 0), torch.cat(label, 0), path, shapes