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