Modify augmentations.py so that we have an augmentation that applies the watershed technique to the images. It will return a tuple of the mask of the images as well the number of focci detected.
NOTE: Watershed augmentation must be the last augmentation to be called as it will return a tuple instead of just the image itself so it will not fit as a input to other augmentations which only expect the image and not a tuple.
Files
augmentation.py
Tasks
[ ] Create new ApplyWatershed class in augmentations.py
[ ] Create a call function for augmentations.py and apply watershed to the image get the number of focci and also return the masked image for later
Description
Modify augmentations.py so that we have an augmentation that applies the watershed technique to the images. It will return a tuple of the mask of the images as well the number of focci detected.
NOTE: Watershed augmentation must be the last augmentation to be called as it will return a tuple instead of just the image itself so it will not fit as a input to other augmentations which only expect the image and not a tuple.
Files
augmentation.py
Tasks