TypeError: Can't broadcast (128, 200704) -> (128, 1024, 14, 14)

[liuziyi219]

Hi，I'm getting this error TypeError: Can't broadcast (128, 200704) -> (128, 1024, 14, 14) when running extract_features.py. The error is at line 133 output_h5["features"][counter : counter + len(batch)] = output_features.cpu().numpy()

[deluzhao]

Not sure if you still need this (hopefully this could be helpful for others), but I believe that you can simply reshape the output_features to (128, 1024, 14, 14).

EX: After the error line, insert a "output_features = np.reshape(output_features, (128, 1024, 14, 14))" seems to work for me. Let me know if this runs into any other issues.