Fix quantizise (#26)

* Compare new model 1300 * fix quantization
jomjol · Apr 11, 2024 · 51c8d83 · 51c8d83
1 parent 9357b16
commit 51c8d83
Show file tree

Hide file tree

Showing 4 changed files with 1,856 additions and 2,228 deletions.
diff --git a/Compare_all_tflite.ipynb b/Compare_all_tflite.ipynb
diff --git a/Train_CNN_Ana-Class100_S1.ipynb b/Train_CNN_Ana-Class100_S1.ipynb
@@ -1649,7 +1649,7 @@
     "FileName = TFliteNamingAndVersion + \"_q\" + \".tflite\"\n",
     "\n",
     "def representative_dataset():\n",
-    "    for n in range(x_data[0].size):\n",
+    "    for n in range(np.size(x_data,0)):\n",
     "      data = np.expand_dims(x_data[n], axis=0)\n",
     "      yield [data.astype(np.float32)]\n",
     "        \n",

diff --git a/Train_CNN_Ana-Class100_S2.ipynb b/Train_CNN_Ana-Class100_S2.ipynb
diff --git a/Train_CNN_Ana-Cont-S2.ipynb b/Train_CNN_Ana-Cont-S2.ipynb