segment-anything / download_models.sh
hogepodge's picture
Initial commit of the Label Studio Segment Anything space
6307f85
raw
history blame contribute delete
No virus
603 Bytes
#!/bin/bash
MODELS_DIR="models"
mkdir -p ${MODELS_DIR}
download_model() {
FILE_PATH="${MODELS_DIR}/$1"
URL="$2"
if [ ! -f "${FILE_PATH}" ]; then
wget -q "${URL}" -P ${MODELS_DIR}/
fi
}
# Model files and their corresponding URLs
declare -A MODELS
# We just run with MobileSAM for this example
# MODELS["sam_vit_h_4b8939.pth"]="https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth"
MODELS["mobile_sam.pt"]="https://github.com/ChaoningZhang/MobileSAM/raw/master/weights/mobile_sam.pt"
for model in "${!MODELS[@]}"; do
download_model "${model}" "${MODELS[${model}]}"
done