想要训练edvr,在mmediting项目中进行训练
但是reds数据集太大下载不下来,我们可以选择使用稍小一点的数据集进行训练,看看流程是否走得通
1、修改文件路径
修改配置文件中的数据集路径:
修改该文件内lq和gt和ann文件的路径:
2、生成anno文件
首先训练数据除了 低分辨率、ground truth数据、还需要一个anno数据。这个数据通过该目录下的py文件来获取
修改文件中range的数量,根据数据集的序列个数来修改即可。
之后添加–root-path参数开始运行即可,便可以得到meta_info_REDS_GT.txt文件。
3、修改数据集的具体参数
进入对应文件夹
修改对应的参数即可
4、报错
之后在开始训练(多卡)时会遇到这样的错误
告诉你模型的部分计算没有存放在计算图中,所以需要重新设置参数:
进入对应文件:
在改文件下,将参数find_unused_parameters设为True:
5、训练
一切就绪,开始训练!