OpenMMLab AI实战营 note3

OpenMMLab AI实战营 note3

北京超算使用MMClassfication进行花卉分类

创建环境

根据 mmclassification 的环境要求，需要用 anaconda、cuda、gcc 等基础环境模块

数据集

flower 数据集包含 5 种类别的花卉图像:雏菊 daisy 588张，蒲公英 dandelion 556张，玫瑰 rose 583张，向日葵 sunflower 536张，郁金香 tulip 585张。

数据集下载链接:

国际网:https://www.dropbox.com/s/snom6v4zfky0flx/flower_dataset.zip?dl=0

国内网:https://pan.baidu.com/s/1RJmAoxCD_aNPyTRX6w97xQ 提取码: 9x5u

划分数据集

将数据集按照 8:2 的比例划分成训练和验证子数据集，并将数据集整理成 ImageNet的格式将训练子集和验证子集放到 train 和 val 文件夹下。

使用split_data.py程序

编写MMCls的配置文件

根据官方文档学习即可