Skip to content

GitLab

Commit 2e13b9a7 authored by syiming's avatar syiming
Browse files

change the default max_fpn_level to 6

parent 70c974d3
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
research/object_detection/models/faster_rcnn_resnet_v1_fpn_keras_feature_extractor.py
View file @ 2e13b9a7
...@@ -45,7 +45,7 @@ class FasterRCNNResnetV1FpnKerasFeatureExtractor( ...@@ -45,7 +45,7 @@ class FasterRCNNResnetV1FpnKerasFeatureExtractor(
batch_norm_trainable=False, batch_norm_trainable=False,
weight_decay=0.0, weight_decay=0.0,
fpn_min_level=2, fpn_min_level=2,
fpn_max_level=7, fpn_max_level=6,
additional_layer_depth=256, additional_layer_depth=256,
override_base_feature_extractor_hyperparams=False): override_base_feature_extractor_hyperparams=False):
"""Constructor. """Constructor.
...@@ -256,7 +256,7 @@ class FasterRCNNResnet50FpnKerasFeatureExtractor( ...@@ -256,7 +256,7 @@ class FasterRCNNResnet50FpnKerasFeatureExtractor(
batch_norm_trainable=False, batch_norm_trainable=False,
weight_decay=0.0, weight_decay=0.0,
fpn_min_level=2, fpn_min_level=2,
fpn_max_level=7, fpn_max_level=6,
additional_layer_depth=256, additional_layer_depth=256,
override_base_feature_extractor_hyperparams=False): override_base_feature_extractor_hyperparams=False):
"""Constructor. """Constructor.
...@@ -296,7 +296,7 @@ class FasterRCNNResnet101FpnKerasFeatureExtractor( ...@@ -296,7 +296,7 @@ class FasterRCNNResnet101FpnKerasFeatureExtractor(
batch_norm_trainable=False, batch_norm_trainable=False,
weight_decay=0.0, weight_decay=0.0,
fpn_min_level=2, fpn_min_level=2,
fpn_max_level=7, fpn_max_level=6,
additional_layer_depth=256, additional_layer_depth=256,
override_base_feature_extractor_hyperparams=False): override_base_feature_extractor_hyperparams=False):
"""Constructor. """Constructor.
...@@ -337,7 +337,7 @@ class FasterRCNNResnet152FpnKerasFeatureExtractor( ...@@ -337,7 +337,7 @@ class FasterRCNNResnet152FpnKerasFeatureExtractor(
batch_norm_trainable=False, batch_norm_trainable=False,
weight_decay=0.0, weight_decay=0.0,
fpn_min_level=2, fpn_min_level=2,
fpn_max_level=7, fpn_max_level=6,
additional_layer_depth=256, additional_layer_depth=256,
override_base_feature_extractor_hyperparams=False): override_base_feature_extractor_hyperparams=False):
"""Constructor. """Constructor.
......
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