Commit 0e1082f5 authored by Teodors EGLITIS's avatar Teodors EGLITIS

Changes to resolve Merge Request 17

parent 834fd6bf
Pipeline #5101 passed with stages
in 16 minutes and 7 seconds
from .utils import ManualRoiCut from .utils import ManualRoiCut
from .utils import ConstructVeinImage from .utils import ConstructVeinImage
from .utils import RotateImage from .utils import NormalizeImageRotation
# gets sphinx autodoc done right - don't remove it # gets sphinx autodoc done right - don't remove it
__all__ = [_ for _ in dir() if not _.startswith('_')] __all__ = [_ for _ in dir() if not _.startswith('_')]
This diff is collapsed.
...@@ -231,7 +231,7 @@ def test_ConstructAnnotations(): ...@@ -231,7 +231,7 @@ def test_ConstructAnnotations():
image_filename = F( ( 'preprocessors', 'ConstructAnnotations.png' ) ) image_filename = F( ( 'preprocessors', 'ConstructAnnotations.png' ) )
roi_annotations_filename = F( ( 'preprocessors', 'ConstructAnnotations.txt' ) ) roi_annotations_filename = F( ( 'preprocessors', 'ConstructAnnotations.txt' ) )
vein_annotations_filename = F( ( 'preprocessors', 'ConstructAnnotations.npy' ) ) vein_annotations_filename = F( ( 'preprocessors', 'ConstructAnnotations.npy' ) )
image = bob.io.base.load( image_filename ) image = bob.io.base.load( image_filename )
roi_annotations = np.loadtxt(roi_annotations_filename, dtype='uint16') roi_annotations = np.loadtxt(roi_annotations_filename, dtype='uint16')
roi_annotations = [tuple([point[0], point[1]]) for point in roi_annotations] roi_annotations = [tuple([point[0], point[1]]) for point in roi_annotations]
...@@ -243,9 +243,9 @@ def test_ConstructAnnotations(): ...@@ -243,9 +243,9 @@ def test_ConstructAnnotations():
annotation_dictionary = {"image" : image, "roi_annotations" : roi_annotations, "vein_annotations" : vein_annotations} annotation_dictionary = {"image" : image, "roi_annotations" : roi_annotations, "vein_annotations" : vein_annotations}
from bob.bio.vein.preprocessor.utils import ConstructVeinImage from bob.bio.vein.preprocessor.utils import ConstructVeinImage
from bob.bio.vein.preprocessor.utils import RotateImage from bob.bio.vein.preprocessor.utils import NormalizeImageRotation
output = ConstructVeinImage(annotation_dictionary, center = True).return_annotations() output = ConstructVeinImage(annotation_dictionary, center = True)
output = RotateImage(output, dark_lines = False).rotate() output = NormalizeImageRotation(output, dark_lines = False)
assert np.array_equal(output, image) assert np.array_equal(output, image)
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment