williamyang1991 / VToonify

[SIGGRAPH Asia 2022] VToonify: Controllable High-Resolution Portrait Video Style Transfer
Other
3.53k stars 446 forks source link

where is pretrain.pt ? #44

Closed undcloud closed 1 year ago

undcloud commented 1 year ago

Hello, sir , I meet this issue:

FileNotFoundError: [Errno 2] No such file or directory: './checkpoint/vtoonify_d_cartoon/pretrain.pt'

I can't find pretrain.pt in google drive:

|--vtoonify_t_cartoon
    |--pretrain.pt                        % * Pre-trained encoder for Cartoon style
|--vtoonify_d_cartoon
    |--pretrain.pt                        % * Pre-trained encoder for Cartoon style

Could you tell me where is pretrain.pt? thanks~

undcloud commented 1 year ago

Should I pretrain first, after that I get a pretrain.pt?

williamyang1991 commented 1 year ago

Yes. pretrain.pt is not required for testing. So I didn't release pretrain.pt. If you want to train your model, you need first pretrain the model to get pretrain.pt.

undcloud commented 1 year ago

Thanks~,I have trained the model, I have seen the good effect.