updates
This commit is contained in:
+1
-1
@@ -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
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user