-
-
Notifications
You must be signed in to change notification settings - Fork 44
Expand file tree
/
Copy pathconvert_all_meta_sam2.sh
More file actions
49 lines (41 loc) · 2.46 KB
/
convert_all_meta_sam2.sh
File metadata and controls
49 lines (41 loc) · 2.46 KB
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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
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
echo "Converting SAM2.1-Hiera-Tiny models..."
python -m samexporter.export_sam2 --checkpoint "original_models/sam2.1_hiera_tiny.pt" \
--output_encoder "output_models/sam2.1_hiera_tiny.encoder.onnx" \
--output_decoder "output_models/sam2.1_hiera_tiny.decoder.onnx" \
--model_type sam2.1_hiera_tiny
echo "Converting SAM2.1-Hiera-Small models..."
python -m samexporter.export_sam2 --checkpoint "original_models/sam2.1_hiera_small.pt" \
--output_encoder "output_models/sam2.1_hiera_small.encoder.onnx" \
--output_decoder "output_models/sam2.1_hiera_small.decoder.onnx" \
--model_type sam2.1_hiera_small
echo "Converting SAM2.1-Hiera-BasePlus models..."
python -m samexporter.export_sam2 --checkpoint "original_models/sam2.1_hiera_base_plus.pt" \
--output_encoder "output_models/sam2.1_hiera_base_plus.encoder.onnx" \
--output_decoder "output_models/sam2.1_hiera_base_plus.decoder.onnx" \
--model_type sam2.1_hiera_base_plus
echo "Converting SAM2.1-Hiera-Large models..."
python -m samexporter.export_sam2 --checkpoint "original_models/sam2.1_hiera_large.pt" \
--output_encoder "output_models/sam2.1_hiera_large.encoder.onnx" \
--output_decoder "output_models/sam2.1_hiera_large.decoder.onnx" \
--model_type sam2.1_hiera_large