support keras functional model and tuning fallback layers (#521)

Signed-off-by: Clark Chin <[email protected]> Co-authored-by: chensuyue <[email protected]> Co-authored-by: Lv, Liang1 <[email protected]>
intel · Mar 2, 2023 · efd7379 · efd7379
1 parent 4bd540c
commit efd7379
Show file tree

Hide file tree

Showing 48 changed files with 25,228 additions and 1,372 deletions.
diff --git a/examples/.config/model_params_keras.json b/examples/.config/model_params_keras.json
@@ -0,0 +1,39 @@
+{
+  "keras": {
+    "resnetv2_50": {
+      "model_src_dir": "image_recognition/resnetv2_50/quantization/ptq",
+      "dataset_location": "/tf_dataset/dataset/imagenet",
+      "input_model": "/tf_dataset2/models/tensorflow/resnetv2_50_keras/saved_model",
+      "main_script": "main.py",
+      "batch_size": 32
+    },
+    "xception": {
+      "model_src_dir": "image_recognition/xception/quantization/ptq",
+      "dataset_location": "/tf_dataset/dataset/imagenet",
+      "input_model": "/tf_dataset2/models/tensorflow/xception_keras/saved_model/",
+      "main_script": "main.py",
+      "batch_size": 32
+    },
+    "mnist": {
+      "model_src_dir": "image_recognition/mnist/quantization/qat",
+      "dataset_location": "",
+      "input_model": "/tf_dataset2/models/tensorflow/mnist_keras/saved_model/",
+      "main_script": "main.py",
+      "batch_size": 32
+    },
+    "resnetv2_101": {
+      "model_src_dir": "image_recognition/resnetv2_101/quantization/ptq",
+      "dataset_location": "/tf_dataset/dataset/imagenet",
+      "input_model": "/tf_dataset2/models/tensorflow/resnetv2_101_keras/saved_model",
+      "main_script": "main.py",
+      "batch_size": 32
+    },
+    "mobilenet_v2": {
+      "model_src_dir": "image_recognition/mobilenet_v2/quantization/ptq",
+      "dataset_location": "/tf_dataset/dataset/imagenet",
+      "input_model": "/tf_dataset2/models/tensorflow/mobilenet_v2_keras/saved_model/",
+      "main_script": "main.py",
+      "batch_size": 32
+    }
+  }
+}
diff --git a/examples/README.md b/examples/README.md
@@ -61,7 +61,7 @@ Intel® Neural Compressor validated examples with multiple compression technique
     <td>MobileNet V2</td>
     <td>Image Recognition</td>
     <td>Post-Training Static Quantization</td>
-    <td><a href="./tensorflow/image_recognition/tensorflow_models/mobilenet_v2/quantization/ptq">pb</a> / <a href="./tensorflow/image_recognition/keras_models/mobilenet_v2/quantization/ptq">keras</a></td>
+    <td><a href="./tensorflow/image_recognition/tensorflow_models/mobilenet_v2/quantization/ptq">pb</a> / <a href="./keras/image_recognition/mobilenet_v2/quantization/ptq">keras</a></td>
   </tr>
   <tr>
     <td>MobileNet V3</td>
@@ -115,13 +115,13 @@ Intel® Neural Compressor validated examples with multiple compression technique
     <td>ResNet V2 50</td>
     <td>Image Recognition</td>
     <td>Post-Training Static Quantization</td>
-    <td><a href="./tensorflow/image_recognition/tensorflow_models/resnet_v2_50/quantization/ptq">pb</a> / <a href="./tensorflow/image_recognition/keras_models/resnetv2_50/quantization/ptq">keras</a></td>
+    <td><a href="./tensorflow/image_recognition/tensorflow_models/resnet_v2_50/quantization/ptq">pb</a> / <a href="./keras/image_recognition/resnetv2_50/quantization/ptq">keras</a></td>
   </tr>
   <tr>
     <td>ResNet V2 101</td>
     <td>Image Recognition</td>
     <td>Post-Training Static Quantization</td>
-    <td><a href="./tensorflow/image_recognition/tensorflow_models/resnet_v2_101/quantization/ptq">pb</a> / <a href="./tensorflow/image_recognition/keras_models/resnetv2_101/quantization/ptq">keras</a></td>
+    <td><a href="./tensorflow/image_recognition/tensorflow_models/resnet_v2_101/quantization/ptq">pb</a> / <a href="./keras/image_recognition/resnetv2_101/quantization/ptq">keras</a></td>
   </tr>
   <tr>
     <td>ResNet V2 152</td>
@@ -193,7 +193,7 @@ Intel® Neural Compressor validated examples with multiple compression technique
     <td>Xception</td>
     <td>Image Recognition</td>
     <td>Post-Training Static Quantization</td>
-    <td><a href="./tensorflow/image_recognition/keras_models/xception/quantization/ptq">keras</a></td>
+    <td><a href="./keras/image_recognition/xception/quantization/ptq">keras</a></td>
   </tr>
   <tr>
     <td>ResNet V2</td>