Commit ef68a826 authored by syiming's avatar syiming
Browse files

fix error in get_box_classifier_feature_extractor_model

parent 7a147514
...@@ -218,8 +218,9 @@ class FasterRCNNResnetV1FPNKerasFeatureExtractor( ...@@ -218,8 +218,9 @@ class FasterRCNNResnetV1FPNKerasFeatureExtractor(
with tf.name_scope(name): with tf.name_scope(name):
with tf.name_scope('ResnetV1FPN'): with tf.name_scope('ResnetV1FPN'):
feature_extractor_model = tf.keras.models.Sequential([ feature_extractor_model = tf.keras.models.Sequential([
tf.keras.layers.Dense(unit=1024, activation='ReLU'), tf.keras.layers.Flatten(),
tf.keras.layers.Dense(unit=1024, activation='ReLU') tf.keras.layers.Dense(units=1024, activation='relu'),
tf.keras.layers.Dense(units=1024, activation='relu')
]) ])
return feature_extractor_model return feature_extractor_model
......
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