Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity

Edit comment (#3759)

Browse Source 
edit comment
This commit is contained in:
Yonghye Kwon authored and GitHub committed 2021-06-24 15:57:27 +02:00
1 parent ae4261c774
commit 417a2f425c
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
utils/datasets.py
+1 -1
View File
@@ -574,7 +574,7 @@ class LoadImagesAndLabels(Dataset): # for training/testing
labels_out[:, 1:] = torch.from_numpy(labels) labels_out[:, 1:] = torch.from_numpy(labels)
# Convert # Convert
img = img[:, :, ::-1].transpose(2, 0, 1) # BGR to RGB, to 3x416x416 img = img[:, :, ::-1].transpose(2, 0, 1) # BGR to RGB, to 3 x img_height x img_width
img = np.ascontiguousarray(img) img = np.ascontiguousarray(img)
return torch.from_numpy(img), labels_out, self.img_files[index], shapes return torch.from_numpy(img), labels_out, self.img_files[index], shapes