Home

Awesome

LoRASAM

This repository contains the code for Low-Rank Adaptation of Segment Anything Model for Surgical Scene Segmentation

Environment File

Create a new conda environment with the config file given in the repository as follows:

conda env create -f lorasam.yml
conda activate lorasam

General file descriptions

Example Usage for Training

python driver_scratchpad.py --model_config model_loratuning.yml --data_config config_endovis.yml --save_path "./temp.pth"

Please refer to driver_scratchpad.py for other command line options and parameters.

Example Usage for Evaluation

cd eval/endovis

bash generate_all_predictions.sh

Citation

To be added

Please feel free to reach out to me or raise an issue in case of trouble while running the code.