search

yuke93 / RL-Restore

Crafting a Toolchain for Image Restoration by Deep Reinforcement Learning (CVPR 2018 Spotlight)
MIT License
210 stars 48 forks source link

dataset question #7

Closed wanghai111 closed 5 years ago

wanghai111 commented 5 years ago

How to make my data set as a training set？

yuke93 commented 5 years ago

You may refer to data/train/generate_train.m, which is used to generate training data.

wanghai111 commented 5 years ago

It can be made using the dataset you provide, but my own dataset will show insufficient memory.

yuke93 commented 5 years ago

It seems your dataset is large. You may divide your dataset into some subsets, and you can generate one hdf5 file for each subset.

wanghai111 commented 5 years ago

I am using the DIV2K800 dataset, your link to download the DIV2K dataset can be made successfully, but the DIV2K dataset I downloaded is not successful.

yuke93 commented 5 years ago

Yes. I have down-sampled the original images by different scales, so the original images have larger sizes.