| --- |
| 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. |
|
|
