How to genereate the pt file which required by DIV2K_train.txt?

[leiwen83]

Hi,

For training process, it require, the pt format, while the original DIV2k is in png format. Is there some script which could be used to convert?

Thanks,

[NJU-Jet]

import imageio import pickle

with open('0.pt', 'wb') as f: pickle.dump(imageio.imread('0.png'), f)

[leiwen83]

Got it.