diff options
Diffstat (limited to 'src/argaze/AreaOfInterest/AOI3DScene.py')
-rw-r--r-- | src/argaze/AreaOfInterest/AOI3DScene.py | 8 |
1 files changed, 3 insertions, 5 deletions
diff --git a/src/argaze/AreaOfInterest/AOI3DScene.py b/src/argaze/AreaOfInterest/AOI3DScene.py index d126f45..28150c6 100644 --- a/src/argaze/AreaOfInterest/AOI3DScene.py +++ b/src/argaze/AreaOfInterest/AOI3DScene.py @@ -87,7 +87,7 @@ class AOI3DScene(AOIFeatures.AOIScene): # retreive all aoi3D vertices for name, face in faces.items(): - aoi3D = AOIFeatures.AreaOfInterest(vertices=[ vertices[i-1] for i in face ]) + aoi3D = numpy.array([ vertices[i-1] for i in face ]).astype('float32').view(AOIFeatures.AreaOfInterest) self.append(name, aoi3D) except IOError: @@ -101,12 +101,10 @@ class AOI3DScene(AOIFeatures.AOIScene): for name, aoi3D in self.areas.items(): - vertices_3D = numpy.array(aoi3D.vertices).astype('float32') - - vertices_2D, J = cv.projectPoints(vertices_3D, self.rotation, self.translation, K, D) + vertices_2D, J = cv.projectPoints(aoi3D, self.rotation, self.translation, K, D) vertices_2D = vertices_2D.astype('int').reshape((len(vertices_2D), 2)).tolist() - aoi2D = AOIFeatures.AreaOfInterest(vertices_2D) + aoi2D = numpy.array(vertices_2D).view(AOIFeatures.AreaOfInterest) aoi2D_scene.append(name, aoi2D) |