add keras models

Signed-off-by: zehao-intel <[email protected]>

examples/.config/model_params_keras_3x.json

@@ -0,0 +1,25 @@
+{
+  "keras": {
+    "resnetv2_50": {
+      "model_src_dir": "keras/cv/resnet_v2_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
+    },
+    "inception_v3": {
+      "model_src_dir": "keras/cv/inception_v3/quantization/ptq",
+      "dataset_location": "/tf_dataset/dataset/imagenet",
+      "input_model": "/tf_dataset2/models/tensorflow/inception_v3_keras/saved_model",
+      "main_script": "main.py",
+      "batch_size": 32
+    },
+    "mobilenet_v2": {
+      "model_src_dir": "keras/cv/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
+    }
+  }
+}

examples/.config/model_params_tensorflow_3x.json

@@ -72,6 +72,13 @@
       "main_script": "main.py",
       "batch_size": 32
     },
+    "resnetv2_50": {
+      "model_src_dir": "cv/resnet_v2_50/quantization/ptq",
+      "dataset_location": "/tf_dataset/dataset/imagenet",
+      "input_model": "/tf_dataset/pre-train-model-slim/pbfile/frozen_pb/frozen_resnet_v2_50.pb",
+      "main_script": "main.py",
+      "batch_size": 32
+    },
     "vgg16": {
       "model_src_dir": "cv/vgg16/quantization/ptq",
       "dataset_location": "/tf_dataset/dataset/imagenet",

examples/3.x_api/tensorflow/cv/mobilenet_v2/quantization/ptq/main.py

@@ -117,7 +117,6 @@ def run(self):
             q_model.save(args.output_graph)
 
         if args.benchmark:
-            dataloader = create_dataloader('tensorflow', dataloader_args)
             dataset = ImageRecordDataset(
                 root=args.dataset_location, 
                 transform=ComposeTransform(transform_list= [