$ hub install dpn131_imagenet==1.0.0
$ hub run dpn131_imagenet --input_path "/PATH/TO/IMAGE"
$ hub run dpn131_imagenet --input_file test.txt
test.txt 存放待分类图片的存放路径
def classification(data)
用于图像分类预测
参数
data:dict类型,key为image,str类型,value为待检测的图片路径,list类型。
返回
result:list类型,每个元素为对应输入图片的预测结果。预测结果为dict类型,key为该图片分类结果label,value为该label对应的概率
预测代码示例
import paddlehub as hub
module = hub.Module(name="dpn131_imagenet")
test_img_path = "path/to/image"
# 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)
https://github.com/PaddlePaddle/models/tree/develop/PaddleCV/image_classification
paddlepaddle >= 1.4.0
paddlehub >= 1.0.0