-
-
Notifications
You must be signed in to change notification settings - Fork 36
/
convert_all_meta_sam2.sh
25 lines (21 loc) · 1.21 KB
/
convert_all_meta_sam2.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
echo "Converting all models..."
echo "Converting SAM2-Hiera-Tiny models..."
python -m samexporter.export_sam2 --checkpoint original_models/sam2_hiera_tiny.pt \
--output_encoder output_models/sam2_hiera_tiny.encoder.onnx \
--output_decoder output_models/sam2_hiera_tiny.decoder.onnx \
--model_type sam2_hiera_tiny
echo "Converting SAM2-Hiera-Small models..."
python -m samexporter.export_sam2 --checkpoint original_models/sam2_hiera_small.pt \
--output_encoder output_models/sam2_hiera_small.encoder.onnx \
--output_decoder output_models/sam2_hiera_small.decoder.onnx \
--model_type sam2_hiera_small
echo "Converting SAM2-Hiera-BasePlus models..."
python -m samexporter.export_sam2 --checkpoint original_models/sam2_hiera_base_plus.pt \
--output_encoder output_models/sam2_hiera_base_plus.encoder.onnx \
--output_decoder output_models/sam2_hiera_base_plus.decoder.onnx \
--model_type sam2_hiera_base_plus
echo "Converting SAM2-Hiera-Large models..."
python -m samexporter.export_sam2 --checkpoint original_models/sam2_hiera_large.pt \
--output_encoder output_models/sam2_hiera_large.encoder.onnx \
--output_decoder output_models/sam2_hiera_large.decoder.onnx \
--model_type sam2_hiera_large