Same problem in 2 Questions, 1 using Keras, 1 using PyTorch -
Using a pre-trained PyTorch model (NCHW format), I converted it to Ke