mirror of
https://github.com/wassname/sloth.git
synced 2026-07-11 01:24:41 +08:00
remove obsolete loaders
This commit is contained in:
@@ -1,80 +0,0 @@
|
||||
import os
|
||||
|
||||
class Loader:
|
||||
def __init__(self):
|
||||
pass
|
||||
|
||||
def load(self, filename):
|
||||
pass
|
||||
|
||||
def save(self, filename, annotations):
|
||||
pass
|
||||
|
||||
class RectIdLoader(Loader):
|
||||
def load(self, filename):
|
||||
annotations = []
|
||||
|
||||
print "loading ", filename
|
||||
filename_helper = {}
|
||||
|
||||
f = open(filename)
|
||||
basedir = ""
|
||||
try:
|
||||
for line in f:
|
||||
s = line.split()
|
||||
print s
|
||||
image_fname = s[0]
|
||||
if basedir == "":
|
||||
basedir = os.path.dirname(image_fname)
|
||||
else:
|
||||
assert basedir == os.path.dirname(image_fname)
|
||||
num_rects = int(s[1])
|
||||
assert num_rects == 0 or num_rects == 1
|
||||
if image_fname not in filename_helper:
|
||||
annotation = {
|
||||
'type': 'image',
|
||||
'filename': image_fname,
|
||||
'annotations': []
|
||||
}
|
||||
annotations.append(annotation)
|
||||
filename_helper[image_fname] = annotation
|
||||
|
||||
rects = filename_helper[image_fname]['annotations']
|
||||
|
||||
for i in range(0, 4*num_rects, 4):
|
||||
rects.append({
|
||||
'type': 'rect',
|
||||
'x': s[i+2],
|
||||
'y': s[i+3],
|
||||
'width': s[i+4],
|
||||
'height': s[i+5]
|
||||
})
|
||||
if len(s) > i+6:
|
||||
rects[-1]['id'] = s[i+6]
|
||||
except Exception, e:
|
||||
print e
|
||||
|
||||
print annotations
|
||||
return annotations
|
||||
|
||||
class FeretLoader(Loader):
|
||||
def load(self, filename):
|
||||
basedir = os.path.dirname(filename)
|
||||
f = open(filename)
|
||||
|
||||
annotations = []
|
||||
for line in f:
|
||||
s = line.split()
|
||||
fileitem = {
|
||||
'filename': os.path.join(basedir, s[0]+".bmp"),
|
||||
'type': 'image',
|
||||
}
|
||||
fileitem['annotations'] = [
|
||||
{'type': 'point', 'class': 'left_eye', 'x': int(s[1]), 'y': int(s[2])},
|
||||
{'type': 'point', 'class': 'right_eye', 'x': int(s[3]), 'y': int(s[4])},
|
||||
{'type': 'point', 'class': 'mouth', 'x': int(s[5]), 'y': int(s[6])}
|
||||
]
|
||||
annotations.append(fileitem)
|
||||
|
||||
return annotations
|
||||
|
||||
Reference in New Issue
Block a user