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( ...@@ -215,6 +215,7 @@ class FasterRCNNResnetV1FpnKerasFeatureExtractor(
""" """
with tf.name_scope(name): with tf.name_scope(name):
with tf.name_scope('ResnetV1FPN'): with tf.name_scope('ResnetV1FPN'):
# TODO: Add a batchnorm layer between two fc layers.
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'),
......
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