Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
VVJia authored Aug 10, 2024
1 parent 395cbb2 commit a30c11d
Showing 1 changed file with 24 additions and 0 deletions.
24 changes: 24 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,24 @@ SAM-FNet: SAM-Guided Fusion Network for Laryngo-Pharyngeal Tumor Detection



## Fine-tune SAM with LoRA

To fine-tune SAM using LoRA, we recommend following the guidelines provided in the original repository: [SAMed](https://github.com/hitachinsk/SAMed/tree/main).

### Steps:

1. **Fine-tune SAM:**
- Follow the instructions in the [SAMed repository](https://github.com/hitachinsk/SAMed/tree/main) to fine-tune SAM with LoRA.
2. **Generate Local Images:**
- After fine-tuning, modify the `crop_image.py` script to suit your requirements.
- Run the script to generate local images:

```markdown
python crop_image.py
```



## Dataset

Organize your datasets in the following manner:
Expand Down Expand Up @@ -68,3 +86,9 @@ python train.py --data_dir <Your folder> --save_path <Your output path> --num_cl
```markdown
python val.py --model_path <Your model path> --encoder ResNet50 --dataset <Your dataset name> --save_path <Your output path>
```



## Acknowledgement

The code of SAM-FNet is built upon [SAMed](https://github.com/hitachinsk/SAMed/tree/main) and [DLGNet](https://github.com/soleilssss/DLGNet), and we express our gratitude to these awesome projects.

0 comments on commit a30c11d

Please sign in to comment.