assafshocher / ZSSR

"Zero-Shot" Super-Resolution using Deep Internal Learning
Other
401 stars 102 forks source link

Custom dataset training #23

Open Kavit212 opened 4 years ago

Kavit212 commented 4 years ago

Hi, thanks for making your great work available to everyone. Is it possible to train the model with my own custom dataset ? If so, could you kindly give me some clues on which lines of codes need to be changed to make this possible?

Thank you very much

assafshocher commented 4 years ago

Hi, Thanks for the kind words! No datasets here, no pretraining. ZSSR trains on the test image, at test time only. So the "dataset" is actually the input image you put in the test directory. Thanks and good luck!

Kavit212 commented 4 years ago

Thanks for the fast reply !

xiaoyi-st commented 3 years ago

嗨,感谢您为所有人提供出色的工作。 是否可以使用我自己的自定义数据集训练模型? 如果是这样,您能为我提供一些有关哪些代码行需要更改以使其实现的线索吗?

非常感谢你

嗨,感谢您为所有人提供出色的工作。 是否可以使用我自己的自定义数据集训练模型? 如果是这样,您能为我提供一些有关哪些代码行需要更改以使其实现的线索吗?

非常感谢你

Hello, have you successfully trained

xiaoyi-st commented 3 years ago

嗨,感谢您为所有人提供出色的工作。 是否可以使用我自己的自定义数据集训练模型? 如果是这样,您能为我提供一些有关哪些代码行需要更改以使其实现的线索吗?

非常感谢你

Hello, did you train successfully