MMEditing代码阅读笔记二:main()函数中的build_dataset()

MMEditing代码阅读笔记二:main()函数中的build_dataset()

今天,继上篇文章捋清楚build_model()函数之后,进入build_dataset()函数里面看了看。

总结就是:换汤不换药。

build_dataset()函数执行框图

在这里插入图片描述最后pipe_line()那块:其实这块从大框架来说并不能当作是一个独立的模块。或者可以这么说(上篇文章 我们说道build_from_cfg函数中的obj_type = args.pop(‘type’)语句会将除了type类型的其他属性copy到args这个参数中,然后args会赋值给type对应value的实例化对像。) 然后,进入pipe_line里面继续迭代取出type,并将其他属性赋值给type对应的实例化对象。 我觉得简单一点就像一个大函数里面有一对小函数,顺序执行罢了。

这里你可以在调试的时候观察type类型的变化 并对比配置文件中的train_pipeline(以edsr_x2c64b16_g1_300k_div2k.py为例) 只是截取了一部分。
在这里插入图片描述

首先,你可以把下图当作大函数,或许叫做外层对象更恰当。
在这里插入图片描述然后,实例化好外层对象之后,开始进入内层对象的建立(对应上面咱说的所谓的小函数,注:对应着上面的字典列表来看,也正是在这一步进入了pipe_line里面) pipe_line的功能就是对数据进行归一化、翻转、取出patch… 也就是常说的预处理。
在这里插入图片描述在这里插入图片描述

到此,退出build_dataset()函数,创建数据部分执行完毕。。
另外,发现train_loader 封装数据这块 在后续的train_model()函数中执行。

好了,关于这部分的记录到此为止。

猜你喜欢

转载自blog.csdn.net/MZYYZT/article/details/129868862