DeniJsonC / WaveNet

[PG 2023] WaveNet: Wave-Aware Image Enhancement
Other
78 stars 14 forks source link

自定义数据集的训练和测试 #6

Open Solitude1024 opened 5 months ago

Solitude1024 commented 5 months ago

作者你好,请问可以使用自定义数据集进行训练和测试吗?如果可以的话,在python项目中需要对哪些代码和设置进行修改呢?

DeniJsonC commented 5 months ago

作者你好,请问可以使用自定义数据集进行训练和测试吗?如果可以的话,在python项目中需要对哪些代码和设置进行修改呢?

你好,如果是成对数据集的话,以LOL数据集为例,参考./Enhancement/Options/*_LOL.yml文件中 将以下路径修改成自己数据集的路径就行了,测试集也如此
Train: dataroot_gt: dataset/LOLdataset/our485/high/ #真实值 dataroot_lq: dataset/LOLdataset/our485/low/ #低照度图像 val: dataroot_gt: dataset/LOLdataset/eval15/high/ #真实值 dataroot_lq: dataset/LOLdataset/eval15/low/ #低照度图像