Implementation Details of Unet-Segmentation-Jax Permalink
Published:
Image segmentation is a fundamental computer vision task that assigns a class label to each pixel, enabling models to separate objects from their backgrounds with pixel-level precision. This blog provides the explanation of a JAX-based U-Net implementation for pet image segmentation using the Oxford-IIIT Pet dataset. The project starts by building a U-Net from scratch, also covering evaluation metrics such as Dice score and Intersection over Union (IoU). It also implements ResNet-based transfer learning as an encoder alternative to compare a fully trained-from-scratch U-Net with a pretrained feature-based segmentation model. Corresponding repository can be found here repository.
