This repository contains a simple PyTorch implementation of an U-Net for semantic segmentation of fish images, using this dataset by B. J. Boom, P. X. Huang and J. He, R. B. Fisher [1].
Here is a sample fish image and its ground truth mask:
The model is very simple and not super accurate, but the results are kinda cute:
The code for the U-Net is partially based on this Kaggle kernel.