Merge remote-tracking branch 'origin/master'
This commit is contained in:
commit
f2cb840123
|
@ -17,7 +17,7 @@ This directory contains python software and an iOS App developed by Ultralytics
|
|||
|
||||
# Description
|
||||
|
||||
The https://github.com/ultralytics/yolov3 repo contains inference and training code for YOLOv3 in PyTorch. The code works on Linux, MacOS and Windows. Training is done on the COCO dataset by default: https://cocodataset.org/#home. **Credit to Joseph Redmon for YOLO: ** https://pjreddie.com/darknet/yolo/.
|
||||
The https://github.com/ultralytics/yolov3 repo contains inference and training code for YOLOv3 in PyTorch. The code works on Linux, MacOS and Windows. Training is done on the COCO dataset by default: https://cocodataset.org/#home. **Credit to Joseph Redmon for YOLO:** https://pjreddie.com/darknet/yolo/.
|
||||
|
||||
# Requirements
|
||||
|
||||
|
|
|
@ -100,7 +100,7 @@ class LoadImagesAndLabels(Dataset): # for training/testing
|
|||
assert len(self.img_files) > 0, 'No images found in %s' % path
|
||||
self.img_size = img_size
|
||||
self.augment = augment
|
||||
self.label_files = [x.replace('images', 'labels').replace('.bmp', '.txt').replace('.jpg', '.txt')
|
||||
self.label_files = [x.replace('images', 'labels').replace('.bmp', '.txt').replace('.jpg', '.txt').replace('.png', '.txt')
|
||||
for x in self.img_files]
|
||||
|
||||
def __len__(self):
|
||||
|
|
Loading…
Reference in New Issue