This commit is contained in:
Glenn Jocher
2019-12-13 18:52:08 -08:00
parent a4bdb8ce2e
commit 0465500b37
24 changed files with 148 additions and 122431 deletions
+1 -1
View File
@@ -282,7 +282,7 @@ class LoadImagesAndLabels(Dataset): # for training/testing
# Rectangular Training https://github.com/ultralytics/yolov3/issues/232
if self.rect:
# Read image shapes
sp = 'data' + os.sep + path.replace('.txt', '.shapes').split(os.sep)[-1] # shapefile path
sp = path.replace('.txt', '.shapes') # shapefile path
try:
with open(sp, 'r') as f: # read existing shapefile
s = [x.split() for x in f.read().splitlines()]
+1 -1
View File
@@ -658,7 +658,7 @@ def coco_class_count(path='../coco/labels/train2014/'):
print(i, len(files))
def coco_only_people(path='../coco/labels/val2014/'):
def coco_only_people(path='../coco/labels/train2017/'): # from utils.utils import *; coco_only_people()
# Find images with only people
files = sorted(glob.glob('%s/*.*' % path))
for i, file in enumerate(files):