使用PaddleHub进行图像分类
图像分类是计算机视觉的重要领域,目标是将图像分类到预定义的标签。本文以Kaggle的猫狗分类数据集为例子,来做PaddleHub图像分类任务。
PaddleHub可以使用关键字进行检索模型匹配,找到自己需要的预训练模型
'''
通过关键字在服务端检索匹配的模型,以关键字resnet为例
'''
!hub search resnet
step1 加载预训练模型
猫狗分类属于图像分类的任务,这里使用经典的ResNet-50作为预训练模型,也就是上面通过关键字搜索的第一个模型。此外还可使用别的模型,如最新的神经网络架构搜索类的PNASNet。
import paddlehub as hub#选择paddlehub中的模型
module=hub.Module(name="resnet_v2_50_imagenet")#加载paddlehub中的resnet_v2_50_imagenet模型
input_dict,outpit_dict,program=module.context(
trainable=True)#获取模型的上下文