def test_mxnet(image):
import mxnet as mx
from xception import Xception
image = np.transpose(image, (0, 3, 1, 2))
net = Xception(1)
net.load_parameters('xception.params')
pred = net(mx.nd.array(image)).asnumpy()
print('mxnet :', pred.argmax())
