Closed ghost closed 6 years ago
https://github.com/alterzero/DBPN-Pytorch/blob/c206139dcc87258fd93fef3bd8d7351bbcd523df/main.py#L33-L34 Hello, can you offer me the augmented datasets or augmented method? thanks you! @alterzero
right, i notice that.
How to train on your own data set?What should be paid attention to..?
@alterzero Hello, can you provide the augment code to pre-process the DIV2K dataset, I can not reproduce the result you report and I think the pre-processing step matters
There is an augment function in dataset.py. Based on the code it seems you may have to run that first to create the augmented dataset and save in separate location. Then you can train on augmented data. I haven't done this so I'm unsure if I'm correct.
@BecketXu Have you solved your problem? It seems that some kind of data augmentation code needs to be executed to produce the training data. Is the code in this project? Or somewhere else?
@BecketXu @jshermeyer DBPN-Pytorch is based on EDSR-Pytorch code. The author of EDSR-Pytorch explained about data argument. Please refer to the following link. https://github.com/thstkdgus35/EDSR-PyTorch/issues/38
@kuroyanagi33 Thank you. I think this solved my problem.
Thanks @kuroyanagi33
I think this was my confusion, from main.py:
parser.add_argument('--data_augmentation', type=bool, default=True) parser.add_argument('--hr_train_dataset', type=str, default='DIV2KHRaug') parser.add_argument('--train_dataset', type=str, default='DIV2KLRaug_x8')
But, if I just point these at non-augmented LR and HR DIV2K and leave the data_augmentation tag set to True, I believe the code will automatically augment during training?
@jshermeyer This code carries out online augmentation (data is not increased during training, randomly inverted, rotated etc only) by data_augmentation = True
. According to personal guesses, I think that 'DIV2K_HR_aug'
was data augmented in some other way (not included in this code). Probably in order to increase accuracy with the NTIRE2018 competition, I think the author of DBPN made data augmentation in advance. The code of online augmentation is based on EDSR-pytorch's author's code. EDSR is the state of the art code before DBPN. The author of EDSR mentioned that the data of DIV2K is so big that he performed a process like online augmentation.
@kuroyanagi33 thanks for explain. i am training the code just with the "augment" func in dataset.py
Hi all, I'm so sorry for my very late reply. I have been swamped with other works. The built-in augmentation function does not contain resize function. So, basically you need to do resize operation offline using matlab to generate more images.
From this code, the authors train the DBPN with the augmented DIV2K dataset. @BecketXu