Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
ResNet50_tensorflow
Commits
8b45de4f
Commit
8b45de4f
authored
May 10, 2021
by
Vighnesh Birodkar
Committed by
TF Object Detection Team
May 10, 2021
Browse files
Parse deepmac meta arch correctly.
PiperOrigin-RevId: 372953929
parent
7f0ee4cb
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
15 additions
and
0 deletions
+15
-0
research/object_detection/builders/model_builder.py
research/object_detection/builders/model_builder.py
+15
-0
No files found.
research/object_detection/builders/model_builder.py
View file @
8b45de4f
...
@@ -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
(
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment