Awesome

Cleavage-Stage Embryo Segmentation Using SAM-Based Dual Branch Pipeline: Development and Evaluation with the CleavageEmbryo Dataset

This is the official pytorch implementation of SAM-Based Dual Branch Pipeline, please refer the paper for more details.

Dataset

CleavageEmbryo Dataset.

Install

The code requires python>=3.8, as well as pytorch>=1.7 and torchvision>=0.8. Install Segment Anything:

pip install git+https://github.com/facebookresearch/segment-anything.git

Install requirements:

pip install -r requirements.txt

QuickStart

sam_dual yolo

Inference test images

python inference.py

Evaluation with Pre-trained Models

python evaluate.py

Training on your own dataset

download sam modify config.py and then

python train_es.py

Citation

Acknowledgment

The code is built on segment-anything-model and lightning-sam, many thanks for the Third Party Libs.