"vscode:/vscode.git/clone" did not exist on "24cde76a152fbffde30fa2be0d08dcbad490530e"
Commit 2e9c6046 authored by Vighnesh Birodkar's avatar Vighnesh Birodkar Committed by TF Object Detection Team
Browse files

Standardize CenterNet mobilenet v2 FPN with new checkpoint API.

PiperOrigin-RevId: 383469336
parent fa038f86
...@@ -125,14 +125,8 @@ class CenterNetMobileNetV2FPNFeatureExtractor( ...@@ -125,14 +125,8 @@ class CenterNetMobileNetV2FPNFeatureExtractor(
self._base_model.load_weights(path) self._base_model.load_weights(path)
@property @property
def supported_sub_model_types(self): def classification_backbone(self):
return ['classification']
def get_sub_model(self, sub_model_type):
if sub_model_type == 'classification':
return self._base_model return self._base_model
else:
ValueError('Sub model type "{}" not supported.'.format(sub_model_type))
def call(self, inputs): def call(self, inputs):
return [self._feature_extractor_model(inputs)] return [self._feature_extractor_model(inputs)]
......
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