pytorch-PCN icon indicating copy to clipboard operation
pytorch-PCN copied to clipboard

How to train on new dataset

Open AllenDun opened this issue 5 years ago • 1 comments

Hi, nice work. How about retraining on a new dataset? Because I find that the pre-trained PCN performs badly on low-resolution images.

AllenDun avatar Feb 26 '19 03:02 AllenDun

I haven't train on new data myself too. I just convert the model from the original repo. For training details, you can refer to the paper and implement by yourself. I have no plan to implement training phrase right now. The author of this model have said he may release the training code in the future, See this

siriusdemon avatar Feb 26 '19 03:02 siriusdemon