RCAN icon indicating copy to clipboard operation
RCAN copied to clipboard

Add my own dataset

Open lovepiano opened this issue 6 years ago • 1 comments

Hi, I don't understand how to add my own dataset in terms of your README file.My dataset has HR and LR, I have no idea about how to add them to your code. Could you please help me? Thanks! Ruan

lovepiano avatar Jun 04 '19 13:06 lovepiano

Hi, I don't understand how to add my own dataset in terms of your README file.My dataset has HR and LR, I have no idea about how to add them to your code. Could you please help me? Thanks! Ruan

If you want to test on your own dataset , you just need HR.However,if you want to use your dataset to train the model, you need HR and LR. Besides, you need to change the '--dir_data',which can be seen in option.py. If you don't use DIV2K to train, Maybe you need to read the div2k.py and write your own dataloader. Or you can replace the DIV2K pictures with your own pictures ,and just replace the DIV2K rgb_mean with your own dataset's rgb_mean, which is in rcan.py

wangyifenggg avatar Jun 09 '19 15:06 wangyifenggg