fix error in get_box_classifier_feature_extractor_model

ef68a826 · syiming · 7a147514 · ef68a826
Commit ef68a826 authored Jun 20, 2020 by syiming
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

research/object_detection/models/faster_rcnn_resnet_v1_fpn_keras_feature_extractor.py ...dels/faster_rcnn_resnet_v1_fpn_keras_feature_extractor.py +3 -2

No files found.
--- a/research/object_detection/models/faster_rcnn_resnet_v1_fpn_keras_feature_extractor.py
+++ b/research/object_detection/models/faster_rcnn_resnet_v1_fpn_keras_feature_extractor.py
@@ -218,8 +218,9 @@ class FasterRCNNResnetV1FPNKerasFeatureExtractor(
    with tf.name_scope(name):
      with tf.name_scope('ResnetV1FPN'):
        feature_extractor_model = tf.keras.models.Sequential([
-          tf.keras.layers.Dense(unit=1024, activation='ReLU'),
-          tf.keras.layers.Dense(unit=1024, activation='ReLU')
+          tf.keras.layers.Flatten(),
+          tf.keras.layers.Dense(units=1024, activation='relu'),
+          tf.keras.layers.Dense(units=1024, activation='relu')
        ])
        return feature_extractor_model