xref: /aosp_15_r20/external/executorch/examples/mediatek/shell_scripts/export_oss.sh (revision 523fa7a60841cd1ecfb9cc4201f1ca8b03ed023a)
1model=$1
2
3echo "Export model: $model"
4
5if [ $model = "deeplabv3" ]
6then
7	python3 model_export_scripts/deeplab_v3.py -d
8elif [ $model = "edsr" ]
9then
10	python3 model_export_scripts/edsr.py -d
11elif [ $model = "inceptionv3" ]
12then
13	python3 model_export_scripts/inception_v3.py -d PATH_TO_DATASET
14elif [ $model = "inceptionv4" ]
15then
16	python3 model_export_scripts/inception_v4.py -d PATH_TO_DATASET
17elif [ $model = "mobilenetv2" ]
18then
19	python3 model_export_scripts/mobilenet_v2.py -d PATH_TO_DATASET
20elif [ $model = "mobilenetv3" ]
21then
22	python3 model_export_scripts/mobilenet_v3.py -d PATH_TO_DATASET
23elif [ $model = "resnet18" ]
24then
25	python3 model_export_scripts/resnet18.py -d PATH_TO_DATASET
26elif [ $model = "resnet50" ]
27then
28	python3 model_export_scripts/resnet50.py -d PATH_TO_DATASET
29fi
30