search

raywzy / Bringing-Old-Films-Back-to-Life

Bringing Old Films Back to Life (CVPR 2022)
http://raywzy.com/Old_Film/
513 stars 51 forks source link

How to inference/use the model after training? I got the models but don't know how to use them yet... #10

Open HelixNGC7293 opened 1 year ago

HelixNGC7293 commented 1 year ago

Hello! After training, I got several model files in my OUTPUT folder, such as: net_D_xxx.pth & net_G_xxx.pth & optimizer_xxx.pth

For the next step, how to use these model to colorize/denoise my black&white footage?
(I've converted my footage to be png sequence frame images and put them into a folder already) Any tips will be helpful!! Thanks!!

raywzy commented 1 year ago

The test scripts have been updated, please try and see there are any issues.

HelixNGC7293 commented 1 year ago

0089

Got it working! It's very very impressive! Thank you so much! Btw is there any tutorial/test code for the colorization part of this project? I can manage to manually colorizing the keyframes by hands, just don't know how to import them in this project yet...

~~Additionally, how to change the resolution for final output? Currently the height is always 368px, when I tried to change it, it pops error like: RuntimeError: grid_sampler(): expected grid and input to have same batch size, but got input with sizes [87210, 1, 51, 90] and grid with sizes [85500, 9, 9, 2]~~ Got it working, the image size needs to be a multiple of 16.

makemota commented 1 year ago

Have the same question - After training, I got 288 model files in my OUTPUT/RNN_Swin_4/models folder, such as: 96 files net_D_xxx.pth & 96 files net_G_xxx.pth & 96 files optimizer_xxx.pth How to use these models? The test scripts have been updated.