Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleClas
提交
d1d818f5
P
PaddleClas
项目概览
PaddlePaddle
/
PaddleClas
大约 1 年 前同步成功
通知
115
Star
4999
Fork
1114
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
19
列表
看板
标记
里程碑
合并请求
6
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleClas
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
19
Issue
19
列表
看板
标记
里程碑
合并请求
6
合并请求
6
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
d1d818f5
编写于
11月 04, 2021
作者:
C
cuicheng01
提交者:
GitHub
11月 04, 2021
浏览文件
操作
浏览文件
下载
差异文件
Merge pull request #1391 from cuicheng01/develop
add quick_start_multilabel_classification.md
上级
4dcf7ffb
f785932d
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
109 addition
and
2 deletion
+109
-2
docs/zh_CN/quick_start/quick_start_multilabel_classification.md
...h_CN/quick_start/quick_start_multilabel_classification.md
+108
-0
docs/zh_CN/quick_start/quick_start_recognition.md
docs/zh_CN/quick_start/quick_start_recognition.md
+1
-2
未找到文件。
docs/zh_CN/quick_start/quick_start_multilabel_classification.md
0 → 100644
浏览文件 @
d1d818f5
# 多标签分类quick start
基于
[
NUS-WIDE-SCENE
](
https://lms.comp.nus.edu.sg/wp-content/uploads/2019/research/nuswide/NUS-WIDE.html
)
数据集,体验多标签分类的训练、评估、预测的过程,该数据集是NUS-WIDE数据集的一个子集。请首先安装PaddlePaddle和PaddleClas,具体安装步骤可详看
[
Paddle 安装文档
](
../installation/install_paddle.md
)
,
[
PaddleClas 安装文档
](
../installation/install_paddleclas.md
)
。
## 目录
*
[
数据和模型准备
](
#1
)
*
[
模型训练
](
#2
)
*
[
模型评估
](
#3
)
*
[
模型预测
](
#4
)
*
[
基于预测引擎预测
](
#5
)
*
[
5.1 导出inference model
](
#5.1
)
*
[
5.2 基于预测引擎预测
](
#5.2
)
<a
name=
"1"
></a>
## 一、数据和模型准备
*
进入PaddleClas目录。
```
cd path_to_PaddleClas
```
*
创建并进入
`dataset/NUS-WIDE-SCENE`
目录,下载并解压NUS-WIDE-SCENE数据集。
```
shell
mkdir
dataset/NUS-WIDE-SCENE
cd
dataset/NUS-WIDE-SCENE
wget https://paddle-imagenet-models-name.bj.bcebos.com/data/NUS-SCENE-dataset.tar
tar
-xf
NUS-SCENE-dataset.tar
```
*
返回
`PaddleClas`
根目录
```
cd ../../
```
<a
name=
"2"
></a>
## 二、模型训练
```
shell
export
CUDA_VISIBLE_DEVICES
=
0,1,2,3
python3
-m
paddle.distributed.launch
\
--gpus
=
"0,1,2,3"
\
tools/train.py
\
-c
./ppcls/configs/quick_start/professional/MobileNetV1_multilabel.yaml
```
训练10epoch之后,验证集最好的正确率应该在0.95左右。
<a
name=
"3"
></a>
## 三、模型评估
```
bash
python3 tools/eval.py
\
-c
./ppcls/configs/quick_start/professional/MobileNetV1_multilabel.yaml
\
-o
Arch.pretrained
=
"./output/MobileNetV1/best_model"
```
<a
name=
"4"
></a>
## 四、模型预测
```
bash
python3 tools/infer.py
\
-c
./ppcls/configs/quick_start/professional/MobileNetV1_multilabel.yaml
\
-o
Arch.pretrained
=
"./output/MobileNetV1/best_model"
```
得到类似下面的输出:
```
[{'class_ids': [6, 13, 17, 23, 26, 30], 'scores': [0.95683, 0.5567, 0.55211, 0.99088, 0.5943, 0.78767], 'file_name': './deploy/images/0517_2715693311.jpg', 'label_names': []}]
```
<a
name=
"5"
></a>
## 五、基于预测引擎预测
<a
name=
"5.1"
></a>
### 5.1 导出inference model
```
bash
python3 tools/export_model.py
\
-c
./ppcls/configs/quick_start/professional/MobileNetV1_multilabel.yaml
\
-o
Arch.pretrained
=
"./output/MobileNetV1/best_model"
```
inference model的路径默认在当前路径下
`./inference`
<a
name=
"5.2"
></a>
### 5.2 基于预测引擎预测
首先进入deploy目录下:
```
bash
cd
./deploy
```
通过预测引擎推理预测:
```
python3 python/predict_cls.py \
-c configs/inference_multilabel_cls.yaml
```
得到类似下面的输出:
```
0517_2715693311.jpg: class id(s): [6, 13, 17, 23, 26, 30], score(s): [0.96, 0.56, 0.55, 0.99, 0.59, 0.79], label_name(s): []
```
docs/zh_CN/quick_start/quick_start_recognition.md
浏览文件 @
d1d818f5
...
...
@@ -74,8 +74,7 @@ cd ..
wget
{
数据下载链接地址
}
&&
tar
-xf
{
压缩包的名称
}
```
<a
name=
"下载、解压inference_模型与demo数据"
></a>
<a
name=
"下载、解压inference模型与demo数据"
></a>
### 2.1 下载、解压 inference 模型与 demo 数据
下载 demo 数据集以及轻量级主体检测、识别模型,命令如下。
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录