third_party/opencv
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Improvements in tests, bugfix in digits_video sample

Browse Source
This commit is contained in:
Vladislav Sovrasov
2016-02-05 17:46:52 +03:00
parent 4e3a6328ba
commit 87fc75c6d7
8 changed files with 62 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/python/test/tst_scene_render.py
+6 -2
View File
@@ -13,9 +13,10 @@ defaultSize = 512
class TestSceneRender():
def __init__(self, bgImg = None, **params):
def __init__(self, bgImg = None, deformation = False, **params):
self.time = 0.0
self.timeStep = 1.0 / 30.0
self.deformation = deformation
if bgImg != None:
self.sceneBg = bgImg.copy()
@@ -26,7 +27,7 @@ class TestSceneRender():
self.h = self.sceneBg.shape[1]
self.initialRect = np.array([ (self.h/2, self.w/2), (self.h/2, self.w/2 + self.w/10),
(self.h/2 + self.h/10, self.w/2 + self.w/10), (self.h/2 + self.h/10, self.w/2)])
(self.h/2 + self.h/10, self.w/2 + self.w/10), (self.h/2 + self.h/10, self.w/2)]).astype(int)
self.currentRect = self.initialRect
def setInitialRect(self, rect):
@@ -42,6 +43,9 @@ class TestSceneRender():
img = self.sceneBg.copy()
self.currentRect = self.initialRect + np.int( 30*cos(self.time) + 50*sin(self.time/3))
if(self.deformation):
self.currentRect[1:3] += np.int(self.h/20*cos(self.time))
cv2.fillConvexPoly(img, self.currentRect, (0, 0, 255))
return img