Skip to content

GitLab

Commit d1d33691 authored by syiming's avatar syiming
Browse files

changing argument sequece to make it consistent with model builder

parent 938f102d
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 6 deletions
research/object_detection/models/faster_rcnn_resnet_v1_fpn_keras_feature_extractor.py
View file @ d1d33691
...@@ -278,8 +278,8 @@ class FasterRCNNResnet50FpnKerasFeatureExtractor( ...@@ -278,8 +278,8 @@ class FasterRCNNResnet50FpnKerasFeatureExtractor(
def __init__(self, def __init__(self,
is_training, is_training,
first_stage_features_stride=16, first_stage_features_stride=16,
conv_hyperparams=None,
batch_norm_trainable=False, batch_norm_trainable=False,
conv_hyperparams=None,
weight_decay=0.0, weight_decay=0.0,
fpn_min_level=2, fpn_min_level=2,
fpn_max_level=6, fpn_max_level=6,
...@@ -290,8 +290,8 @@ class FasterRCNNResnet50FpnKerasFeatureExtractor( ...@@ -290,8 +290,8 @@ class FasterRCNNResnet50FpnKerasFeatureExtractor(
Args: Args:
is_training: See base class. is_training: See base class.
first_stage_features_stride: See base class. first_stage_features_stride: See base class.
conv_hyperparams: See base class.
batch_norm_trainable: See base class. batch_norm_trainable: See base class.
conv_hyperparams: See base class.
weight_decay: See base class. weight_decay: See base class.
fpn_min_level: See base class. fpn_min_level: See base class.
fpn_max_level: See base class. fpn_max_level: See base class.
...@@ -318,8 +318,8 @@ class FasterRCNNResnet101FpnKerasFeatureExtractor( ...@@ -318,8 +318,8 @@ class FasterRCNNResnet101FpnKerasFeatureExtractor(
def __init__(self, def __init__(self,
is_training, is_training,
first_stage_features_stride=16, first_stage_features_stride=16,
conv_hyperparams=None,
batch_norm_trainable=False, batch_norm_trainable=False,
conv_hyperparams=None,
weight_decay=0.0, weight_decay=0.0,
fpn_min_level=2, fpn_min_level=2,
fpn_max_level=6, fpn_max_level=6,
...@@ -330,8 +330,8 @@ class FasterRCNNResnet101FpnKerasFeatureExtractor( ...@@ -330,8 +330,8 @@ class FasterRCNNResnet101FpnKerasFeatureExtractor(
Args: Args:
is_training: See base class. is_training: See base class.
first_stage_features_stride: See base class. first_stage_features_stride: See base class.
conv_hyperparams: See base class.
batch_norm_trainable: See base class. batch_norm_trainable: See base class.
conv_hyperparams: See base class.
weight_decay: See base class. weight_decay: See base class.
fpn_min_level: See base class. fpn_min_level: See base class.
fpn_max_level: See base class. fpn_max_level: See base class.
...@@ -359,8 +359,8 @@ class FasterRCNNResnet152FpnKerasFeatureExtractor( ...@@ -359,8 +359,8 @@ class FasterRCNNResnet152FpnKerasFeatureExtractor(
def __init__(self, def __init__(self,
is_training, is_training,
first_stage_features_stride=16, first_stage_features_stride=16,
conv_hyperparams=None,
batch_norm_trainable=False, batch_norm_trainable=False,
conv_hyperparams=None,
weight_decay=0.0, weight_decay=0.0,
fpn_min_level=2, fpn_min_level=2,
fpn_max_level=6, fpn_max_level=6,
...@@ -371,8 +371,8 @@ class FasterRCNNResnet152FpnKerasFeatureExtractor( ...@@ -371,8 +371,8 @@ class FasterRCNNResnet152FpnKerasFeatureExtractor(
Args: Args:
is_training: See base class. is_training: See base class.
first_stage_features_stride: See base class. first_stage_features_stride: See base class.
conv_hyperparams: See base class.
batch_norm_trainable: See base class. batch_norm_trainable: See base class.
conv_hyperparams: See base class.
weight_decay: See base class. weight_decay: See base class.
fpn_min_level: See base class. fpn_min_level: See base class.
fpn_max_level: See base class. fpn_max_level: See base class.
......
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