ViRefSAM

ViRefSAM for Remote Sensing Few-shot Segmentation

Update: The code will be continuously updated.

Requirements

• Python 3.10
• PyTorch 1.12
• cuda 11.6

Conda environment settings:

conda create -n vrpsam python=3.10
conda activate vrpsam

conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.6 -c pytorch -c conda-forge

Segment-Anything-Model setting:

cd ./segment-anything
pip install -v -e .
cd ..

Preparing Few-Shot Segmentation Datasets

Download following datasets: