"vscode:/vscode.git/clone" did not exist on "40a7b8629e6a0470b7baac2b7843ecc7e923ca93"
Commit 67336612 authored by Vighnesh Birodkar's avatar Vighnesh Birodkar Committed by TF Object Detection Team
Browse files

Add point to convert_keras_models.py in ResNet error message.

PiperOrigin-RevId: 324052434
parent 5366f605
...@@ -129,8 +129,10 @@ class CenterNetResnetFeatureExtractor(CenterNetFeatureExtractor): ...@@ -129,8 +129,10 @@ class CenterNetResnetFeatureExtractor(CenterNetFeatureExtractor):
else: else:
supported_types = ['classification'] supported_types = ['classification']
raise ValueError( raise ValueError(
('Sub model {} is not defined for ResNet.'.format(sub_model_type) + ('Sub model {} is not defined for ResNet.'.format(sub_model_type)
'Supported types are {}.'.format(supported_types))) + 'Supported types are {}.'.format(supported_types)
+ 'Use the script convert_keras_models.py to create your own '
+ 'classification checkpoints.'))
def resnet_v2_101(channel_means, channel_stds, bgr_ordering): def resnet_v2_101(channel_means, channel_stds, bgr_ordering):
......
...@@ -166,7 +166,9 @@ class CenterNetResnetV1FpnFeatureExtractor(CenterNetFeatureExtractor): ...@@ -166,7 +166,9 @@ class CenterNetResnetV1FpnFeatureExtractor(CenterNetFeatureExtractor):
supported_types = ['classification'] supported_types = ['classification']
raise ValueError( raise ValueError(
('Sub model {} is not defined for ResNet FPN.'.format(sub_model_type) ('Sub model {} is not defined for ResNet FPN.'.format(sub_model_type)
+ 'Supported types are {}.'.format(supported_types))) + 'Supported types are {}.'.format(supported_types))
+ 'Use the script convert_keras_models.py to create your own '
+ 'classification checkpoints.')
def resnet_v1_101_fpn(channel_means, channel_stds, bgr_ordering): def resnet_v1_101_fpn(channel_means, channel_stds, bgr_ordering):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment