nouveau modèle

CnnClassification.py

@@ -42,6 +42,6 @@ def load_image(img_path):
     img_array = img_array / 255. 
     return img_array
 
-model_CNN = tf.keras.models.load_model("models/modele_crnn_gpu_final.keras", compile=False)
+model_CNN = tf.keras.models.load_model("models/modele_crnn_hd_final_13genres.keras", compile=False)
 
 print(model_CNN)

convert_model.py

@@ -1,8 +1,8 @@
 import tensorflow as tf
 import os
 
-old_model_path = "models/modele_crnn_gpu_final.h5"
-new_model_path = "models/modele_crnn_gpu_final.keras"
+old_model_path = "models/modele_crnn_hd_final_13genres.h5"
+new_model_path = "models/modele_crnn_hd_final_13genres.keras"
 
 print(f"Chargement de {old_model_path} avec TF {tf.__version__}...")
 

models/modele_crnn_hd_final_13genres.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b6a59354eaf6ec9bf7ba5b693f4212db599cee09c4be0deedbd484f2de43274d
+size 54933600

models/modele_crnn_hd_final_13genres.keras

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5a55f63a16069ab5f6a41180cdccfbecd35e6a1f917a25010932ee40414e859e
+size 18379951