only resnet50

cfbc26cd · sjjsmuel · 287a9b5e · cfbc26cd
Commit cfbc26cd authored Apr 15, 2020 by sjjsmuel
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 4 deletions

classifier/Resnet50.py classifier/Resnet50.py +1 -4

No files found.
--- a/classifier/Resnet50.py
+++ b/classifier/Resnet50.py
@@ -29,10 +29,7 @@ class Resnet50(NetworkBase):
        if shouldSave:
            base_model.save('data/resnet_50_base_model.h5')

-        x = Flatten()(base_model.output)
-        x = Dropout(0.4)(x)
-        x = Dense(128, activation='relu')(x)
-        x = Dropout(0.4)(x)
+        x = GlobalAveragePooling2D(base_model.output)
        out = Dense(self.NUM_CLASSES, activation='softmax', name='probs')(x)

        model = Model(base_model.input, out)