Skip to content

Commit 6293008

Browse files
committed
Update densenet-121-tf to use modern keras API
Ticket 148371
1 parent 3508d77 commit 6293008

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

models/public/densenet-121-tf/README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ For details, see [TensorFlow\* API docs](https://www.tensorflow.org/api_docs/pyt
2525

2626
### Original Model
2727

28-
Image, name: `input_1`, shape: `1, 224, 224, 3`, format: `B, H, W, C`, where:
28+
Image, name: `keras_tensor`, shape: `1, 224, 224, 3`, format: `B, H, W, C`, where:
2929

3030
- `B` - batch size
3131
- `H` - image height
@@ -37,7 +37,7 @@ Mean values - [123.68, 116.78, 103.94], scale values - [58.395,57.12,57.375].
3737

3838
### Converted Model
3939

40-
Image, name: `input_1`, shape: `1, 224, 224, 3`, format: `B, H, W, C`, where:
40+
Image, name: `keras_tensor`, shape: `1, 224, 224, 3`, format: `B, H, W, C`, where:
4141

4242
- `B` - batch size
4343
- `H` - image height

models/public/densenet-121-tf/model.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -25,13 +25,13 @@ files:
2525
original_source: https://storage.googleapis.com/tensorflow/keras-applications/densenet/densenet121_weights_tf_dim_ordering_tf_kernels.h5
2626
source: https://storage.openvinotoolkit.org/repositories/open_model_zoo/public/2022.1/densenet-121-tf/densenet121_weights_tf_dim_ordering_tf_kernels.h5
2727
input_info:
28-
- name: input_1
28+
- name: keras_tensor
2929
shape: [1, 224, 224, 3]
3030
layout: NHWC
3131
model_optimizer_args:
3232
- --reverse_input_channels
33-
- --mean_values=input_1[123.68,116.78,103.94]
34-
- --scale_values=input_1[58.395,57.12,57.375]
33+
- --mean_values=keras_tensor[123.68,116.78,103.94]
34+
- --scale_values=keras_tensor[58.395,57.12,57.375]
3535
- --saved_model_dir=$conv_dir/densenet-121.savedmodel
3636
framework: tf
3737
license: https://raw.githubusercontent.com/tensorflow/tensorflow/master/LICENSE

models/public/densenet-121-tf/pre-convert.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ def main():
3030
model = tf.keras.applications.DenseNet121(
3131
weights=str(args.input_dir / 'densenet121_weights_tf_dim_ordering_tf_kernels.h5')
3232
)
33-
model.save(filepath=args.output_dir / 'densenet-121.savedmodel')
33+
model.export(filepath=args.output_dir / 'densenet-121.savedmodel')
3434

3535

3636
if __name__ == '__main__':

0 commit comments

Comments
 (0)