Commit 8b45de4f authored by Vighnesh Birodkar's avatar Vighnesh Birodkar Committed by TF Object Detection Team
Browse files

Parse deepmac meta arch correctly.

PiperOrigin-RevId: 372953929
parent 7f0ee4cb
...@@ -1056,6 +1056,21 @@ def _build_center_net_model(center_net_config, is_training, add_summaries): ...@@ -1056,6 +1056,21 @@ def _build_center_net_model(center_net_config, is_training, add_summaries):
object_detection_params = object_detection_proto_to_params( object_detection_params = object_detection_proto_to_params(
center_net_config.object_detection_task) center_net_config.object_detection_task)
if center_net_config.HasField('deepmac_mask_estimation'):
logging.warn(('Building experimental DeepMAC meta-arch.'
' Some features may be omitted.'))
deepmac_params = deepmac_meta_arch.deepmac_proto_to_params(
center_net_config.deepmac_mask_estimation)
return deepmac_meta_arch.DeepMACMetaArch(
is_training=is_training,
add_summaries=add_summaries,
num_classes=center_net_config.num_classes,
feature_extractor=feature_extractor,
image_resizer_fn=image_resizer_fn,
object_center_params=object_center_params,
object_detection_params=object_detection_params,
deepmac_params=deepmac_params)
keypoint_params_dict = None keypoint_params_dict = None
if center_net_config.keypoint_estimation_task: if center_net_config.keypoint_estimation_task:
label_map_proto = label_map_util.load_labelmap( label_map_proto = label_map_util.load_labelmap(
......
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