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

add todo in get_box_classifier_feature_extractor_model

parent e4244a46
......@@ -215,6 +215,7 @@ class FasterRCNNResnetV1FpnKerasFeatureExtractor(
"""
with tf.name_scope(name):
with tf.name_scope('ResnetV1FPN'):
# TODO: Add a batchnorm layer between two fc layers.
feature_extractor_model = tf.keras.models.Sequential([
tf.keras.layers.Flatten(),
tf.keras.layers.Dense(units=1024, activation='relu'),
......
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