目录
一、环境搭建
环境说明:
运行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模型 |
预测配置文件 |
构建索引数据库的配置文件 |
通用主体检测模型 |
一般情况 |