Pengxiao-Wang / Style2Paints_V3

Reimplementation of Style2Paints V3

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Where is pre-trained UNet model?

mhcho0110 opened this issue · comments

Hi, I would like to test your model.
but I can't find pre-trained UNet model.
It seems that Unet weights is used in your code in Pytorch-Style2paints/test.py (maybe 25_Unet_checkpoint.pth.tar). Could you share the model?

sorry to be late! Now, u can download "11_Unet_checkpoint.pth.tar" and "11_D_checkpoint.pth.tar" from my BaiduCloud:

链接:https://pan.baidu.com/s/1RVv1XfPv0wUZyhOVT1JSEw 密码:geyn

As long time has passed, the latest weights of Unet has been lost. At present, the weights files at 11th epoch are all I can find. But it still saves you time if you start training with the weights i provided.

the result of the model using "11_Unet_checkpoint.pth.tar" is also shown in BaiduCloud

Hope that's helpful, Best wishes!

您的这个链接失效了欸,您能重新发下不?谢谢~

它弹出来了 hhh