SAM-VMNet / README.md
ly17's picture
Add pipeline tag (#1)
a703418 verified
---
pipeline_tag: mask-generation
---
# SAM-VMNet
SAM-VMNet is a project that combines the power of SAM (Segment Anything Model) and VMNet (Vision-based Medical Network) for advanced medical image segmentation. This project is based on the research paper available at [arXiv:2406.00492](https://arxiv.org/abs/2406.00492).
## Table of Contents
- [Installation](#installation)
- [Usage](#usage)
- [Pre-trained Weights](#pre-trained-weights)
- [License](#license)
- [Citation](#citation)
## Installation
To set up the project, follow these steps:
1. **Clone the repository**:
```bash
git clone https://github.com/qimingfan10/SAM-VMNet.git
cd SAM-VMNet
```
2. **Install the required dependencies**:
```bash
pip install -r requirements.txt
```
## Usage
After installing the dependencies, you can proceed with the following steps:
1. **Download the pre-trained weights**:
- Download the following files from Google Drive and place them in the `SAM-VMNet/VM-UNet/pre_trained_weights` directory:
- [vmamba_small_e238_ema.pth](https://drive.google.com/file/d/1XL7JuacjoZCr8w2b0c8CaQn8b0hREblk/view?usp=drive_link)
- [best-epoch142-loss0.3230.pth](https://drive.google.com/file/d/1OKIzUM_L6FeEqyuIsAMn4x-FHptizTkG/view?usp=drive_link)
- [best-epoch142-loss0.3488.pth](https://drive.google.com/file/d/1jsZKakA4FrYaMXNp6qkVtxXwwcJQKrW4/view?usp=drive_link)
- Download the following file from Google Drive and place it in the `SAM-VMNet/MedSAM-main/work_dir/MedSAM` directory:
- [MedSAM_model.pth](https://drive.google.com/file/d/1O5IVkcVxd2RtOcZEKuTR3WkOBiosHBfz/view?usp=drive_link)
2. **Run the project**:
- Follow the instructions in the project's source code to run the segmentation tasks.
## Pre-trained Weights
The pre-trained weights are essential for the project to function correctly. Ensure that you download and place them in the correct directories as specified in the [Usage](#usage) section.
## License
This project is licensed under the MIT License. See the [LICENSE](LICENSE) file for more details.
## Citation
If you use this project in your research, please cite the following paper:
```bibtex
@article{author2024samvmnet,
title={SAM-VMNet: Advanced Medical Image Segmentation},
author={Author, First and Author, Second},
journal={arXiv preprint arXiv:2406.00492},
year={2024}
}
```
---
For any questions or issues, please open an issue on this repository.