Skip to content

Commit ab11049

Browse files
authored
add layouts for remained public models (#3186)
1 parent 97ac78c commit ab11049

File tree

7 files changed

+7
-5
lines changed

7 files changed

+7
-5
lines changed

models/public/deeplabv3/model.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ model_optimizer_args:
3030
- --reverse_input_channels
3131
- --input_shape=[1,513,513,3]
3232
- --input=1:mul_1
33-
# - --layout=1:mul_1(NHWC)
33+
- --layout=1:mul_1(NHWC)
3434
- --output=ArgMax
3535
- --input_model=$dl_dir/deeplabv3_mnv2_pascal_train_aug/frozen_inference_graph.pb
3636
framework: tf

models/public/efficientdet-d0-tf/model.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,7 @@ postprocessing:
117117
model_optimizer_args:
118118
- --input_shape=[1,512,512,3]
119119
- --input=image_arrays
120-
# - --layout=image_arrays(NHWC)
120+
- --layout=image_arrays(NHWC)
121121
- --reverse_input_channels
122122
- --input_model=$conv_dir/efficientdet-d0_saved_model/efficientdet-d0_frozen.pb
123123
- --transformations_config=$mo_ext_dir/front/tf/automl_efficientdet.json

models/public/efficientdet-d1-tf/model.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,7 @@ postprocessing:
117117
model_optimizer_args:
118118
- --input_shape=[1,640,640,3]
119119
- --input=image_arrays
120-
# - --layout=image_arrays(NHWC)
120+
- --layout=image_arrays(NHWC)
121121
- --reverse_input_channels
122122
- --input_model=$conv_dir/efficientdet-d1_saved_model/efficientdet-d1_frozen.pb
123123
- --transformations_config=$mo_ext_dir/front/tf/automl_efficientdet.json

models/public/efficientnet-b0/model.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ postprocessing:
3232
model_optimizer_args:
3333
- --input_shape=[1,224,224,3]
3434
- --input=0:sub
35-
# - --layout=0:sub(NHWC)
35+
- --layout=0:sub(NHWC)
3636
- --output=logits
3737
- --input_meta_graph=$dl_dir/efficientnet-b0/model.ckpt.meta
3838
- --reverse_input_channels

models/public/efficientnet-v2-b0/model.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,7 @@ conversion_to_onnx_args:
4949
model_optimizer_args:
5050
- --input_shape=[1,3,224,224]
5151
- --input=input
52+
- --layout=input(NCHW)
5253
- --input_model=$conv_dir/efficientnet-v2-b0.onnx
5354
- --mean_values=input[123.675,116.28,103.53]
5455
- --scale_values=input[58.395,57.12,57.375]

models/public/efficientnet-v2-s/model.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,7 @@ model_optimizer_args:
5050
- --input_shape=[1,3,384,384]
5151
- --input=input
5252
- --input_model=$conv_dir/efficientnet-v2-s.onnx
53+
- --layout=input(NCHW)
5354
- --mean_values=input[127.5,127.5,127.5]
5455
- --scale_values=input[127.5,127.5,127.5]
5556
- --reverse_input_channels

models/public/facenet-20180408-102900/model.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ model_optimizer_args:
3333
- --reverse_input_channels
3434
- --input_shape=[1,160,160,3],[1]
3535
- --input=image_batch,phase_train
36-
# - --layout=image_batch(NHWC)
36+
- --layout=image_batch(NHWC)
3737
- --mean_values=image_batch[127.5,127.5,127.5]
3838
- --scale_values=image_batch[128.0]
3939
- --output=embeddings

0 commit comments

Comments
 (0)