Bringing-Old-Films-Back-to-Life

Bringing-Old-Films-Back-to-Life copied to clipboard

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!!

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

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.

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.