训练入口——tools/train.py
核心代码:
1. cfg = Config.fromfile(args.config)——读取配置文件
cfg结构如下:
里面的一些属性都是"ConfigDict"类别
2. 定义数据集——datasets = [build_dataset(cfg.data.train)]
cfg.data.train也是"ConfigDict"类别,如下图,可以看到这个里面的内容其实就是config文件里面train里面的一些参数
build_dataset方法如下:
def build_dataset(cfg, default_args=None):
dataset = build_from_cfg(cfg, DATASETS, default_args)
return dataset
build_from_cfg方法如下,基本上所有的模型、数据等配置都要通过这个方法将配置参数转化为对应的类去使用。