Skip to content
Snippets Groups Projects
Commit 2bbaa2db authored by André Anjos's avatar André Anjos :speech_balloon:
Browse files

Ensures input image has dtype==float64 before anything else

parent a8586e10
No related branches found
No related tags found
1 merge request!353DFV and multiple fixes
...@@ -158,8 +158,6 @@ class MaximumCurvature (Extractor): ...@@ -158,8 +158,6 @@ class MaximumCurvature (Extractor):
image_g2_45 = 0.5*image_g2_0 + fxy + 0.5*image_g2_90 image_g2_45 = 0.5*image_g2_0 + fxy + 0.5*image_g2_90
image_g2_m45 = 0.5*image_g2_0 - fxy + 0.5*image_g2_90 image_g2_m45 = 0.5*image_g2_0 - fxy + 0.5*image_g2_90
img_h, img_w = image.shape #Image height and width
# ###################################################################### # ######################################################################
# [Step 1-1] Calculation of curvature profiles # [Step 1-1] Calculation of curvature profiles
# ###################################################################### # ######################################################################
...@@ -469,7 +467,7 @@ class MaximumCurvature (Extractor): ...@@ -469,7 +467,7 @@ class MaximumCurvature (Extractor):
def __call__(self, image): def __call__(self, image):
finger_image = image[0] finger_image = image[0].astype('float64')
finger_mask = image[1] finger_mask = image[1]
#import time #import time
...@@ -484,8 +482,7 @@ class MaximumCurvature (Extractor): ...@@ -484,8 +482,7 @@ class MaximumCurvature (Extractor):
V = self.eval_vein_probabilities(kappa) V = self.eval_vein_probabilities(kappa)
#self._view_four(V, "Center Probabilities - $V_i$") #self._view_single(V, "Accumulated Probabilities - V")
#self._view_single(V.sum(axis=2), "Accumulated Probabilities - V")
#print('probabilities took %.2f seconds' % (time.time() - start)) #print('probabilities took %.2f seconds' % (time.time() - start))
#start = time.time() #start = time.time()
......
...@@ -125,6 +125,7 @@ def test_max_curvature_HE(): ...@@ -125,6 +125,7 @@ def test_max_curvature_HE():
from bob.bio.vein.extractor.MaximumCurvature import MaximumCurvature from bob.bio.vein.extractor.MaximumCurvature import MaximumCurvature
MC = MaximumCurvature(sigma = 5) MC = MaximumCurvature(sigma = 5)
extr_data = MC(preproc_data) extr_data = MC(preproc_data)
#preprocessor_utils.show_image((255.*extr_data).astype('uint8'))
def test_repeated_line_tracking(): def test_repeated_line_tracking():
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment