百度AI攻略：Paddlehub实现图像分类

【推荐】2019 Java 开发者跳槽指南.pdf(吐血整理) >>>

PaddleHub可以便捷地获取PaddlePaddle生态下的预训练模型，完成模型的管理和一键预测。配合使用Fine-tune API，可以基于大规模预训练模型快速完成迁移学习，让预训练模型能更好地服务于用户特定场景的应用。

本次介绍如何使用paddlehub调用vgg模型实现图像分类。

模型概述

VGG是牛津大学计算机视觉组和DeepMind在2014年提出的一种图像分类模型。该系列模型探索了卷积神经网络的深度与其性能之间的关系，通过实验证明了增加网络的深度能够在一定程度上影响网络最终的性能，到目前为止，VGG仍然被许多其他图像任务用作特征提取的BackBone网络。该PaddleHub Module结构为VGG16，基于ImageNet-2012数据集训练，接受输入图片大小为224 x 224 x 3，支持直接通过命令行或者Python接口进行预测。

module = hub.Module(name="vgg16_imagenet")

test_img_path = "./cat1.jpg"

# 预测结果展示

img = mpimg.imread(test_img_path)

plt.imshow(img)

plt.axis('off')

plt.show()

# set input dict

input_dict = {"image": [test_img_path]}

# execute predict and print the result

results = module.classification(data=input_dict)

for result in results:

print(result)

test_img_path = "./dog1.jpg"

# 预测结果展示

img = mpimg.imread(test_img_path)

plt.imshow(img)

plt.axis('off')

plt.show()

# set input dict

input_dict = {"image": [test_img_path]}

# execute predict and print the result

results = module.classification(data=input_dict)

for result in results:

print(result)

[2020-01-03 09:19:50,058] [ INFO] - Installing vgg16_imagenet module

2020-01-03 09:19:50,058-INFO: Installing vgg16_imagenet module

Downloading vgg16_imagenet

[==================================================] 100.00%

Uncompress /home/aistudio/.paddlehub/cache/vgg16_imagenet

[==================================================] 100.00%

[2020-01-03 09:20:10,875] [ INFO] - Successfully installed vgg16_imagenet-1.0.0

2020-01-03 09:20:10,875-INFO: Successfully installed vgg16_imagenet-1.0.0

[2020-01-03 09:20:11,640] [ INFO] - 32 pretrained paramaters loaded by PaddleHub

2020-01-03 09:20:11,640-INFO: 32 pretrained paramaters loaded by PaddleHub

[{'tiger cat': 0.600113570690155}]

[{'Labrador retriever': 0.9380330443382263}]

整体效果相当不错。

来源：oschina

链接：https://my.oschina.net/u/4438602/blog/3155719

标签

PaddleHub