Fine Tunning SAM on a scientific dataset for segmentation task
Link to the dataset used in this demonstration: https://www.epfl.ch/labs/cvlab/data/data-em/
Source: EPFL
This code has been heavily adapted from this notebook but modified to work with a truly custom dataset where we have a bunch of images and binary masks. https://github.com/NielsRogge/Transformers-Tutorials/blob/master/SAM/Fine_tune_SAM_(segment_anything)_on_a_custom_dataset.ipynb