Okay so you cannot reshape output from this line to this shape. It’s shape should be something like
[16,2,64,64].
so either reshape it to that or make sure your input to extract_features is of the correct shape : [16,12288]