mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-01-18 14:13:15 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

refactor python

Browse Source
This commit is contained in:
marina.kolpakova 2013-01-28 18:41:22 +04:00
parent 8cf30c728e
commit 28098b6632
2 changed files with 21 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
apps/sft/misc/roc_caltech.py
View File

@ -11,23 +11,7 @@ if __name__ == "__main__":
path = "/home/kellan/datasets/caltech/set00/V000.txt"
# open annotation file
f = open(path)
(nFrame, nSample) = sft.caltech.parse_header(f)
objects = sft.caltech.extract_objects(f)
caltechSamples = []
annotations = [[] for i in range(nFrame)]
for obj in objects:
(type, start, end) = re.search(r'^lbl=\'(\w+)\'\s+str=(\d+)\s+end=(\d+)\s+hide=0$', obj[0]).groups()
print type, start, end
start = int(start) -1
end = int(end)
pos = sft.caltech.parse_pos(obj[1])
posv = sft.caltech.parse_pos(obj[2])
occl = sft.caltech.parse_occl(obj[3])
for idx, (p, pv, oc) in enumerate(zip(*[pos, posv, occl])):
annotations[start + idx].append((type, p, oc, pv))
annotations = sft.parse_caltech(f)
for each in annotations:
print each

21
apps/sft/misc/sft.py
View File

@ -259,4 +259,23 @@ class caltech:
def parse_occl(l):
occl = re.match(r'^occl\s*=(\[[\d\s\.\;]+\])$', l).group(1)
occl = re.sub(r"\s(?!\])", ",", occl)
return eval(occl)
return eval(occl)
def parse_caltech(f):
(nFrame, nSample) = caltech.parse_header(f)
objects = caltech.extract_objects(f)
annotations = [[] for i in range(nFrame)]
for obj in objects:
(type, start, end) = re.search(r'^lbl=\'(\w+)\'\s+str=(\d+)\s+end=(\d+)\s+hide=0$', obj[0]).groups()
print type, start, end
start = int(start) -1
end = int(end)
pos = caltech.parse_pos(obj[1])
posv = caltech.parse_pos(obj[2])
occl = caltech.parse_occl(obj[3])
for idx, (p, pv, oc) in enumerate(zip(*[pos, posv, occl])):
annotations[start + idx].append((type, p, oc, pv))
return annotations