From f3cae25c0cb642a7d26ea523f6e9f83f64149712 Mon Sep 17 00:00:00 2001 From: KakeruHirokami <162458293+KakeruHirokami@users.noreply.github.com> Date: Fri, 18 Oct 2024 09:57:25 +0900 Subject: [PATCH] Update download_models.sh --- .../ios/RunScripts/download_models.sh | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/lite/examples/pose_estimation/ios/RunScripts/download_models.sh b/lite/examples/pose_estimation/ios/RunScripts/download_models.sh index a4dfc5760a2..07f67e17f94 100644 --- a/lite/examples/pose_estimation/ios/RunScripts/download_models.sh +++ b/lite/examples/pose_estimation/ios/RunScripts/download_models.sh @@ -26,17 +26,25 @@ if [ ! -f "$FILE" ]; then fi FILE=${MODEL_DIR}/movenet_lightning.tflite +GZ_FILE=${MODEL_DIR}/movenet_lightning.tar.gz if [ ! -f "$FILE" ]; then curl \ - -L 'https://tfhub.dev/google/lite-model/movenet/singlepose/lightning/tflite/float16/4?lite-format=tflite' \ - -o ${FILE} + -L 'https://www.kaggle.com/api/v1/models/google/movenet/tfLite/multipose-lightning-tflite-float16/1/download' \ + -o ${GZ_FILE} + tar -xzf ${GZ_FILE} -C ${MODEL_DIR} + rm ${GZ_FILE} + mv ${MODEL_DIR}/1.tflite ${FILE} fi FILE=${MODEL_DIR}/movenet_thunder.tflite +GZ_FILE=${MODEL_DIR}/movenet_thunder.tar.gz if [ ! -f "$FILE" ]; then curl \ - -L 'https://tfhub.dev/google/lite-model/movenet/singlepose/thunder/tflite/float16/4?lite-format=tflite' \ - -o ${FILE} + -L 'https://www.kaggle.com/api/v1/models/google/movenet/tfLite/singlepose-thunder-tflite-float16/1/download' \ + -o ${GZ_FILE} + tar -xzf ${GZ_FILE} -C ${MODEL_DIR} + rm ${GZ_FILE} + mv ${MODEL_DIR}/4.tflite ${FILE} fi echo -e "Downloaded files are in ${MODEL_DIR}"