OpenMMLab AI实战训练营day5-MMPreTrain代码课

MMPreTrain代码课概述

本次课程主讲人盛入浅出的带我们训练了一个基于resnet18的分类猫狗的二分类分类器，并对训练好的模型进行测试评估，加载训练好的的模型进行推理

数据集下载与划分

首先下载我们需要的数据集，并按照mmpretrain的customdataset格式组织，下图是一个实例。

配置文件的组成

配置文件由四个部分组成，分别是models，datasets,schedules,还有default_runtime

models定义了模型，把模型分成三部分，分别是backbone,neck,head

datasets定义了data_preprocessor,还有数据集的加载器

schedule实现了训练时的策略，包括优化器，学习率调整等

runtime定义了训练时候的一些钩子点位

调用脚本进行训练

可以使用mim命令在终端的任意位置调用mmpretrain的训练脚本，只需要输入下面一行代码