Skip to content

GitLab

Commit bd27aa70 authored by syiming's avatar syiming
Browse files

add batchnorm layer

parent 95d8a879
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
research/object_detection/models/faster_rcnn_resnet_v1_fpn_keras_feature_extractor.py
View file @ bd27aa70
...@@ -295,6 +295,8 @@ class FasterRCNNResnetV1FpnKerasFeatureExtractor( ...@@ -295,6 +295,8 @@ class FasterRCNNResnetV1FpnKerasFeatureExtractor(
feature_extractor_model = tf.keras.models.Sequential([ feature_extractor_model = tf.keras.models.Sequential([
tf.keras.layers.Flatten(), tf.keras.layers.Flatten(),
tf.keras.layers.Dense(units=1024, activation='relu'), tf.keras.layers.Dense(units=1024, activation='relu'),
self._conv_hyperparams.build_batch_norm(
training=(self._is_training and not self._freeze_batchnorm)),
tf.keras.layers.Dense(units=1024, activation='relu'), tf.keras.layers.Dense(units=1024, activation='relu'),
tf.keras.layers.Reshape((1, 1, 1024)) tf.keras.layers.Reshape((1, 1, 1024))
]) ])
......
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