Skip to content

GitLab

Commit 0ea84d6d authored by Ronny Votel's avatar Ronny Votel Committed by TF Object Detection Team
Browse files

Changing the default number of channels for lighter-weight hourglass backbones. 

PiperOrigin-RevId: 363975731
parent 5150eb3e
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 5 deletions
research/object_detection/models/center_net_hourglass_feature_extractor.py
View file @ 0ea84d6d
...@@ -76,7 +76,7 @@ class CenterNetHourglassFeatureExtractor( ...@@ -76,7 +76,7 @@ class CenterNetHourglassFeatureExtractor(
def hourglass_10(channel_means, channel_stds, bgr_ordering): def hourglass_10(channel_means, channel_stds, bgr_ordering):
"""The Hourglass-10 backbone for CenterNet.""" """The Hourglass-10 backbone for CenterNet."""
network = hourglass_network.hourglass_10(num_channels=128) network = hourglass_network.hourglass_10(num_channels=32)
return CenterNetHourglassFeatureExtractor( return CenterNetHourglassFeatureExtractor(
network, channel_means=channel_means, channel_stds=channel_stds, network, channel_means=channel_means, channel_stds=channel_stds,
bgr_ordering=bgr_ordering) bgr_ordering=bgr_ordering)
...@@ -85,16 +85,16 @@ def hourglass_10(channel_means, channel_stds, bgr_ordering): ...@@ -85,16 +85,16 @@ def hourglass_10(channel_means, channel_stds, bgr_ordering):
def hourglass_20(channel_means, channel_stds, bgr_ordering): def hourglass_20(channel_means, channel_stds, bgr_ordering):
"""The Hourglass-20 backbone for CenterNet.""" """The Hourglass-20 backbone for CenterNet."""
network = hourglass_network.hourglass_20(num_channels=128) network = hourglass_network.hourglass_20(num_channels=48)
return CenterNetHourglassFeatureExtractor( return CenterNetHourglassFeatureExtractor(
network, channel_means=channel_means, channel_stds=channel_stds, network, channel_means=channel_means, channel_stds=channel_stds,
bgr_ordering=bgr_ordering) bgr_ordering=bgr_ordering)
def hourglass_32(channel_means, channel_stds, bgr_ordering): def hourglass_32(channel_means, channel_stds, bgr_ordering):
"""The Hourglass-52 backbone for CenterNet.""" """The Hourglass-32 backbone for CenterNet."""
network = hourglass_network.hourglass_32(num_channels=128) network = hourglass_network.hourglass_32(num_channels=48)
return CenterNetHourglassFeatureExtractor( return CenterNetHourglassFeatureExtractor(
network, channel_means=channel_means, channel_stds=channel_stds, network, channel_means=channel_means, channel_stds=channel_stds,
bgr_ordering=bgr_ordering) bgr_ordering=bgr_ordering)
...@@ -103,7 +103,7 @@ def hourglass_32(channel_means, channel_stds, bgr_ordering): ...@@ -103,7 +103,7 @@ def hourglass_32(channel_means, channel_stds, bgr_ordering):
def hourglass_52(channel_means, channel_stds, bgr_ordering): def hourglass_52(channel_means, channel_stds, bgr_ordering):
"""The Hourglass-52 backbone for CenterNet.""" """The Hourglass-52 backbone for CenterNet."""
network = hourglass_network.hourglass_52(num_channels=128) network = hourglass_network.hourglass_52(num_channels=64)
return CenterNetHourglassFeatureExtractor( return CenterNetHourglassFeatureExtractor(
network, channel_means=channel_means, channel_stds=channel_stds, network, channel_means=channel_means, channel_stds=channel_stds,
bgr_ordering=bgr_ordering) bgr_ordering=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