search

marco-c / autowebcompat

Automatically detect web compatibility issues
Mozilla Public License 2.0
35 stars 41 forks source link

VGG16 inconsistent input shape #179

Open sagarvijaygupta opened 6 years ago

sagarvijaygupta commented 6 years ago

With MaxPooling2D with strides, size of image reduces in following manner

(32, 24, 3) (?, 16, 12, 64) (?, 8, 6, 128) (?, 4, 3, 256) (?, 2, 1, 512)

which in next step generates an error

ValueError: Negative dimension size caused by subtracting 2 from 1 for 'max_pooling2d_5/MaxPool' (op: 'MaxPool') with input shapes: [?,2,1,512].

marco-c commented 6 years ago

32x24 might be too small, it was chosen arbitrarily. We should investigate what would be the best size.

sagarvijaygupta commented 6 years ago

Generally for pretrained models we use size of 224x224 or 227x227. We can think about it.