Skip to content
Snippets Groups Projects
Commit 4122ad67 authored by Manuel Günther's avatar Manuel Günther
Browse files

Added implementation for 3D mask extrapolation in preprocessing (not working very well)

parent 8740b92f
No related branches found
No related tags found
No related merge requests found
...@@ -171,7 +171,11 @@ class FaceCrop (Base): ...@@ -171,7 +171,11 @@ class FaceCrop (Base):
if self.mask_sigma is not None: if self.mask_sigma is not None:
# extrapolate the mask so that pixels outside of the image original image region are filled with border pixels # extrapolate the mask so that pixels outside of the image original image region are filled with border pixels
bob.ip.base.extrapolate_mask(self.cropped_mask, cropped_image, self.mask_sigma, self.mask_neighbors, self.mask_rng) if cropped_image.ndim == 2:
bob.ip.base.extrapolate_mask(self.cropped_mask, cropped_image, self.mask_sigma, self.mask_neighbors, self.mask_rng)
else:
[bob.ip.base.extrapolate_mask(self.cropped_mask, cropped_image_channel, self.mask_sigma, self.mask_neighbors, self.mask_rng) for cropped_image_channel in cropped_image]
return cropped_image return cropped_image
......
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