Skip to content

Latest commit

 

History

History
227 lines (144 loc) · 7.67 KB

quick_start.md

File metadata and controls

227 lines (144 loc) · 7.67 KB
Raw

30分钟玩转PaddleClas

基于flowers102数据集，30分钟体验PaddleClas不同骨干网络的模型训练、不同预训练模型、SSLD知识蒸馏方案和数据增广的效果。请事先参考安装指南配置运行环境和克隆PaddleClas代码。

一、数据和模型准备

  • 进入PaddleClas目录。
cd path_to_PaddleClas
  • 进入dataset/flowers102目录，下载并解压flowers102数据集。
cd dataset/flowers102
# 如果希望从浏览器中直接下载，可以复制该链接并访问，然后下载解压即可
wget https://paddle-imagenet-models-name.bj.bcebos.com/data/flowers102.zip
unzip flowers102.zip
  • 返回PaddleClas根目录
cd ../../

二、环境准备

2.1 下载预训练模型

通过下面的命令下载所需要的预训练模型。

mkdir pretrained
cd pretrained
wget https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/ResNet50_vd_pretrained.pdparams
wget https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/ResNet50_vd_ssld_pretrained.pdparams
wget https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/MobileNetV3_large_x1_0_pretrained.pdparams
cd ../

注意：如果是在windows中下载预训练模型的话，需要将地址拷贝到浏览器中下载。

2.2 环境说明

  • 下面所有的训练过程均在单卡V100机器上运行。首先需要设置可用的显卡设备id。

如果使用mac或者linux，可以使用下面的命令进行设置。

export CUDA_VISIBLE_DEVICES=0

如果使用windows，可以使用下面的命令进行设置。

set CUDA_VISIBLE_DEVICES=0
  • 如果希望在cpu上训练，可以将配置文件中的use_gpu: True修改为use_gpu: False，或者在训练脚本后面添加-o use_gpu=False，表示传入参数，覆盖默认的use_gpu值。

三、模型训练

3.1 零基础训练：不加载预训练模型的训练

  • 基于ResNet50_vd模型，训练脚本如下所示。
python3 tools/train.py -c ./configs/quick_start/ResNet50_vd.yaml

如果希望在cpu上训练，训练脚本如下所示。

python3 tools/train.py -c ./configs/quick_start/ResNet50_vd.yaml -o use_gpu=False

下面的训练任务中，如果希望使用cpu训练，也可以在训练脚本中添加-o use_gpu=False。

验证集的Top1 Acc曲线如下所示，最高准确率为0.2735。

3.2 模型微调-基于ResNet50_vd预训练模型(准确率79.12%)

  • 基于ImageNet1k分类预训练模型进行微调，训练脚本如下所示。
python3 tools/train.py -c ./configs/quick_start/ResNet50_vd_finetune.yaml

验证集的Top1 Acc曲线如下所示，最高准确率为0.9402，加载预训练模型之后，flowers102数据集精度大幅提升，绝对精度涨幅超过65%。

使用训练完的预训练模型对图片docs/images/quick_start/flowers102/image_06739.jpg进行预测，预测命令为

python3 tools/infer/infer.py \
    -i docs/images/quick_start/flowers102/image_06739.jpg \
    --model=ResNet50_vd \
    --pretrained_model="output/ResNet50_vd/best_model/ppcls" \
    --class_num=102

最终可以得到如下结果，打印出了Top-5对应的class id以及score。

Current image file: docs/images/quick_start/flowers102/image_06739.jpg
	top1, class id: 0, probability: 0.5129
	top2, class id: 50, probability: 0.0671
	top3, class id: 18, probability: 0.0377
	top4, class id: 82, probability: 0.0238
	top5, class id: 54, probability: 0.0231
  • 注意：这里每个模型的训练结果都不相同，因此结果可能稍有不同。

3.3 SSLD模型微调-基于ResNet50_vd_ssld预训练模型(准确率82.39%)

需要注意的是，在使用通过知识蒸馏得到的预训练模型进行微调时，我们推荐使用相对较小的网络中间层学习率。

ARCHITECTURE:
    name: 'ResNet50_vd'
    params:
        lr_mult_list: [0.5, 0.5, 0.6, 0.6, 0.8]
pretrained_model: "./pretrained/ResNet50_vd_ssld_pretrained"

训练脚本如下。

python3 tools/train.py -c ./configs/quick_start/ResNet50_vd_ssld_finetune.yaml

最终flowers102验证集上精度指标为0.95，相对于79.12%预训练模型的微调结构，新数据集指标可以再次提升0.98%。

3.4 尝试更多的模型结构-MobileNetV3

训练脚本如下所示。

python3 tools/train.py -c ./configs/quick_start/MobileNetV3_large_x1_0_finetune.yaml

最终flowers102验证集上的精度为0.90，比加载了预训练模型的ResNet50_vd的精度差了5%。不同模型结构的网络在相同数据集上的性能表现不同，需要根据预测耗时以及存储的需求选择合适的模型。

3.5 数据增广的尝试-RandomErasing

训练数据量较小时，使用数据增广可以进一步提升模型精度，基于3.3节中的训练方法，结合RandomErasing的数据增广方式进行训练，具体的训练脚本如下所示。

python3 tools/train.py -c ./configs/quick_start/ResNet50_vd_ssld_random_erasing_finetune.yaml

最终flowers102验证集上的精度为0.9627，使用数据增广可以使得模型精度再次提升1.27%。

3.6 知识蒸馏小试牛刀

  • 本小节将尝试使用知识蒸馏技术对MobileNetV3_large_x1_0模型进行训练，使用3.5小节训练得到的ResNet50_vd模型作为蒸馏所用的教师模型，首先将3.5小节训练得到的ResNet50_vd模型保存到指定目录，脚本如下。
cp -r output/ResNet50_vd/best_model/  ./pretrained/flowers102_R50_vd_final/
  • 使用flowers102数据集进行模型蒸馏，为了进一步提提升模型的精度，使用extra_list.txt充当无标签数据，在这里有几点需要注意：
    • extra_list.txt与val_list.txt的样本没有重复，因此可以用于扩充知识蒸馏任务的训练数据。
    • 即使引入了有标签的extra_list.txt中的图像，但是代码中没有使用标签信息，因此仍然可以视为无标签的模型蒸馏。
    • 蒸馏过程中，教师模型使用的预训练模型为flowers102数据集上的训练结果，学生模型使用的是ImageNet1k数据集上精度为75.32%的MobileNetV3_large_x1_0预训练模型。

配置文件中数据数量、模型结构、预训练地址以及训练的数据配置如下：

total_images: 7169
ARCHITECTURE:
    name: 'ResNet50_vd_distill_MobileNetV3_large_x1_0'
pretrained_model:
    - "./pretrained/flowers102_R50_vd_final/ppcls"
    - "./pretrained/MobileNetV3_large_x1_0_pretrained/”
TRAIN:
    file_list: "./dataset/flowers102/train_extra_list.txt"

最终的训练脚本如下所示。

python3 tools/train.py -c ./configs/quick_start/R50_vd_distill_MV3_large_x1_0.yaml

最终flowers102验证集上的精度为0.9647，结合更多的无标签数据，使用教师模型进行知识蒸馏，MobileNetV3的精度涨幅高达6.47%。

3.7 精度一览

  • 下表给出了不同训练yaml文件对应的精度。
配置文件 Top1 Acc
ResNet50_vd.yaml 0.2735
MobileNetV3_large_x1_0_finetune.yaml 0.9000
ResNet50_vd_finetune.yaml 0.9402
ResNet50_vd_ssld_finetune.yaml 0.9500
ResNet50_vd_ssld_random_erasing_finetune.yaml 0.9627
R50_vd_distill_MV3_large_x1_0.yaml 0.9647

下图给出了不同配置文件在迭代过程中的Top1 Acc的精度曲线变化图。

  • 注意：flowers102数据集图片数量较少，因此进行训练时，验证集的精度指标可能会有1%左右的波动。

  • 更多训练及评估流程，请参考开始使用文档