fa2144e06f6157054428231d9f3f83cd5eebc181,scripts/classification/imagenet/demo_imagenet.py,,,#,14

Before Change

    transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
])

img_transformed = nd.zeros((1, 3, 224, 224))
img_transformed[0,:,:,:] = transform_fn(img)
pred = net(img_transformed)

topK = 5
ind = nd.topk(pred, k=topK)[0].astype("int")

After Change

    transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
])

pred = net(img.expand_dims(0))

topK = 5
ind = nd.topk(pred, k=topK)[0].astype("int")

Instances