def _test_remove_reddot(self, img):
f0 = CalibrateRedDotFilter()
f1 = SeparateObjectFilter()
meta_img = MetaImg(img, {})
f0.filter(meta_img)
f1.filter(meta_img)
self.assertLess(np.sum(meta_img.meta['mask'])/255, 40)
def _test_remove_reddot(self, img):
f0 = CalibrateRedDotFilter()
f1 = SeparateObjectFilter()
meta_img = MetaImg(img, {})
f0.filter(meta_img)
f1.filter(meta_img)
self.assertLess(np.sum(meta_img.meta['mask']) / 255, 40)
def test_no_red_dot(self):
im3 = cv2.imread('tests/lena.jpg', cv.CV_LOAD_IMAGE_COLOR)
f1 = SeparateObjectFilter()
meta_img = MetaImg(im3, {})
try:
f1.filter(meta_img)
except:
self.fail()
def test_no_red_dot(self):
im3 = cv2.imread('tests/lena.jpg', cv.CV_LOAD_IMAGE_COLOR)
f1 = SeparateObjectFilter()
meta_img = MetaImg(im3, {})
try:
f1.filter(meta_img)
except:
self.fail()
def test_speed(self):
im3 = cv2.imread('tests/reddot_seed.jpg', cv.CV_LOAD_IMAGE_COLOR)
f1 = SeparateObjectFilter()
t = time()
for i in xrange(5):
meta_img = MetaImg(im3, {})
f1.filter(meta_img)
t2 = time()
self.assertLess(t2 - t, 5)
def test_speed(self):
im3 = cv2.imread('tests/reddot_seed.jpg', cv.CV_LOAD_IMAGE_COLOR)
f1 = SeparateObjectFilter()
t = time()
for i in xrange(5):
meta_img = MetaImg(im3, {})
f1.filter(meta_img)
t2 = time()
self.assertLess(t2-t, 5)
def full_fitlers_test(self):
im3 = cv2.imread('tests/reddot_seed.jpg', cv.CV_LOAD_IMAGE_COLOR)
f1 = CalibrateRedDotFilter({'dims': (4, 4), 'dot_diameter_in_mm': 5})
f2 = SeparateObjectFilter()
t = time()
for i in xrange(5):
meta_img = MetaImg(np.array(im3), {})
f1.filter(meta_img)
f2.filter(meta_img)
t2 = time()
self.assertLess(t2 - t, 5) #at least 2 frames for second
def full_fitlers_test(self):
im3 = cv2.imread('tests/reddot_seed.jpg', cv.CV_LOAD_IMAGE_COLOR)
f1 = CalibrateRedDotFilter({'dims': (4,4),
'dot_diameter_in_mm': 5})
f2 = SeparateObjectFilter()
t = time()
for i in xrange(5):
meta_img = MetaImg(np.array(im3), {})
f1.filter(meta_img)
f2.filter(meta_img)
t2 = time()
self.assertLess(t2-t, 5) #at least 2 frames for second
def test_elipse(self):
im3 = cv2.imread('tests/reddot_seed.jpg', cv.CV_LOAD_IMAGE_COLOR)
f1 = SeparateObjectFilter()
meta_img = MetaImg(im3, {})
f1.filter(meta_img)
self.assertTrue(meta_img.meta.get('ellipsis', False))
def test_no_object(self):
im3 = cv2.imread('tests/no_object.jpg', cv.CV_LOAD_IMAGE_COLOR)
f1 = SeparateObjectFilter()
meta_img = MetaImg(im3, {})
# try:
f1.filter(meta_img)
def test_elipse(self):
im3 = cv2.imread('tests/reddot_seed.jpg', cv.CV_LOAD_IMAGE_COLOR)
f1 = SeparateObjectFilter()
meta_img = MetaImg(im3, {})
f1.filter(meta_img)
self.assertTrue(meta_img.meta.get('ellipsis', False))
def test_no_object(self):
im3 = cv2.imread('tests/no_object.jpg', cv.CV_LOAD_IMAGE_COLOR)
f1 = SeparateObjectFilter()
meta_img = MetaImg(im3, {})
# try:
f1.filter(meta_img)
