-
Notifications
You must be signed in to change notification settings - Fork 0
/
adapter.py
29 lines (22 loc) · 834 Bytes
/
adapter.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
from planar import Vec2, BoundingBox
import numpy as np
from cluster_util import PhysicalObject
import SceneEval
import landmark
objects = []
def adapt(landmarks):
'''takes a scene object and returns a list of lists of objects that form groups'''
landmarkDict = dict()
for l in landmarks:
landmarkDict[l.uuid]=l
o = PhysicalObject(l.uuid,
np.array(l.representation.middle),
np.array(l.representation.rect.min_point),
np.array(l.representation.rect.max_point),
l.uuid)
objects.append(o)
bundles = SceneEval.sceneEval(objects)
for i in bundles:
print i.convert(landmarkDict)
results = [bundle.convert(landmarkDict) for bundle in bundles]
return results