第五十六回 徐宁教使钩镰枪 宋江大破连环马-飞桨图像分类套件PaddleClas初探

飞桨图像分类套件PaddleClas初探

PaddleClas介绍

飞桨图像识别套件PaddleClas是飞桨为工业界和学术界所准备的一个图像识别和图像分类任务的工具集，助力使用者训练出更好的视觉模型和应用落地。官网地址：https://github.com/PaddlePaddle/PaddleClas

编译安装

因为fass-cpu安装有问题，据说需要python3.7版本才行
解决的方法是手动下载代码，然后去掉fass-cpu依赖，编译安装

先下载源代码

git clone https://github.com/PaddlePaddle/PaddleClas

修改paddleclas目录的requirements.txt文件，注释掉里面的faiss-cpu，然后安装库

cd PaddleClas && pip install -r requirements.txt

编译安装PaddleClas

cd PaddleClas && python setup.py install

 最终还要装好faiss-cpu库

pip install faiss-cpu
# pip install paddleclas

PULC分类模型快速体验

下载并解压缩测试图片库

!wget https://paddleclas.bj.bcebos.com/data/PULC/pulc_demo_imgs.zip

!unzip pulc_demo_imgs.zip

使用命令预测：

paddleclas --model_name=person_exists --infer_imgs=pulc_demo_imgs/person_exists/objects365_01780782.jpg

 python代码预测

import paddleclas
model = paddleclas.PaddleClas(model_name="person_exists")
result = model.predict(input_data="pulc_demo_imgs/person_exists/objects365_01780782.jpg")
print(next(result))