b0199525a691c3382eefc3487e82dbdbde78d39b,examples/pretrained_cnn/tutorial_models_vgg19.py,,,#,13

Before Change

img1 = img1 - mean

start_time = time.time()
vgg.eval()
output = vgg(img1)
probs = tf.nn.softmax(output)[0].numpy()
print("  End time : %.5ss" % (time.time() - start_time))
preds = (np.argsort(probs)[::-1])[0:5]

After Change

vgg = tl.models.vgg19(pretrained=True)

img = tl.vis.read_image("data/tiger.jpeg")
img = tl.prepro.imresize(img, (224, 224)).astype(np.float32) / 255

start_time = time.time()
output = vgg(img, is_train=False)
probs = tf.nn.softmax(output)[0].numpy()

Instances