Update bird classifier model download URL

vinlemon · Aug 10, 2024 · 9b01f73 · 9b01f73
1 parent c584d64
commit 9b01f73
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/scripts/download-models.sh b/scripts/download-models.sh
@@ -39,8 +39,13 @@ image_classification_init() {
     curl -sLO "${url}"
   done
 
-  # for custom model downloaded from tf hub
-  curl -sL "https://tfhub.dev/google/lite-model/aiy/vision/classifier/birds_V1/3?lite-format=tflite" -o "lite-model_aiy_vision_classifier_birds_V1_3.tflite"
+  # for custom model downloaded from kaggle
+  bird_model_name="lite-model_aiy_vision_classifier_birds_V1_3"
+  kaggle_tflite_filename="3.tflite"
+  curl -sL "https://www.kaggle.com/api/v1/models/google/aiy/tfLite/vision-classifier-birds-v1/3/download" -o "${bird_model_name}.tar.gz"
+  tar -zxvf "${bird_model_name}.tar.gz"
+  mv "${kaggle_tflite_filename}" "${bird_model_name}.tflite"
+  rm -rf "${bird_model_name}.tar.gz"
 
   popd
 }
@@ -184,4 +189,4 @@ image_embedding_init
 face_detection_init
 audio_classification_init
 text_classification_init
-text_embedding_init
+text_embedding_init