apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

strip string of detection values

Browse Source
This commit is contained in:
stephi 2020-06-08 19:13:54 +02:00 committed by darrenl
parent 7532f1ed87
commit 7cc5241664
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
libs/yolo_io.py
View File

@ -139,7 +139,7 @@ class YoloReader:
def parseYoloFormat(self): def parseYoloFormat(self):
bndBoxFile = open(self.filepath, 'r') bndBoxFile = open(self.filepath, 'r')
for bndBox in bndBoxFile: for bndBox in bndBoxFile:
classIndex, xcen, ycen, w, h = bndBox.split(' ') classIndex, xcen, ycen, w, h = bndBox.strip().split(' ')
label, xmin, ymin, xmax, ymax = self.yoloLine2Shape(classIndex, xcen, ycen, w, h) label, xmin, ymin, xmax, ymax = self.yoloLine2Shape(classIndex, xcen, ycen, w, h)
# Caveat: difficult flag is discarded when saved as yolo format. # Caveat: difficult flag is discarded when saved as yolo format.