autowebcompat icon indicating copy to clipboard operation
autowebcompat copied to clipboard

VGG16 inconsistent input shape

Open sagarvijaygupta opened this issue 7 years ago • 2 comments

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].

sagarvijaygupta avatar May 09 '18 04:05 sagarvijaygupta

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

marco-c avatar May 11 '18 16:05 marco-c

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

sagarvijaygupta avatar May 11 '18 17:05 sagarvijaygupta