PaddleClas的使用教程

目录

一、环境搭建

环境说明：

环境要求

1.1 安装PaddlePaddle环境

快速安装

MacOS安装

Window安装

1.2 安装PaddleClas环境

克隆PaddleClas模型库

安装python依赖库

二、项目配置

说明

2.1 配置inference模型

2.2 下载Demo数据

三、使用步骤

3.1 单张图像识别

3.2 多张图像识别

所有图像的识别结果可视化图像也保存在output文件夹内。

四、自定义图像识别

4.1 准备新的数据和标签

4.2 建立新的索引库

4.3 基于新的索引库进行图像识别

五、参考资料

---

一、环境搭建

环境说明：

运行PaddleClas需要PaddlePaddle 2.1.2或更高版本。可以参考下面的步骤安装PaddlePaddle。

环境要求

• python 3.x
• cuda >= 10.1 (如果使用paddlepaddle-gpu)

• cudnn >= 7.6.4 (如果使用paddlepaddle-gpu)
• nccl >= 2.1.2 (如果使用分布式训练/评估)

• gcc >= 8.2

1.1 安装PaddlePaddle环境

快速安装

MacOS安装

python -m pip install paddlepaddle -i https://mirror.baidu.com/pypi/simple

Window安装

python -m pip install paddlepaddle -i https://mirror.baidu.com/pypi/simple

1.2 安装PaddleClas环境

克隆PaddleClas模型库

git clone https://github.com/PaddlePaddle/PaddleClas.git -b release/2.2


git clone https://gitee.com/paddlepaddle/PaddleClas.git -b release/2.2

安装python依赖库

Python依赖库在requirements.txt中给出，可通过如下命令安装：

pip3 install --upgrade -r requirements.txt -i https://mirror.baidu.com/pypi/simple

二、项目配置

说明

目录：PaddleClas/deploy

2.1 配置inference模型

在目录下新建Models，将模型下载到当前目录解压。

4个方向（Logo、卡通脸、车辆、产品）识别模型的检测模型、测试数据下载地址及对应配置文件地址如下。

型号介绍	推荐场景	inference模型	预测配置文件	构建索引数据库的配置文件
通用主体检测模型	一般情况