Skip to content

GitLab

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
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 8 deletions
research/object_detection/models/center_net_mobilenet_v2_fpn_feature_extractor.py
View file @ 2e9c6046
...@@ -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'] return self._base_model
def get_sub_model(self, sub_model_type):
if sub_model_type == 'classification':
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