-
How to use:
- Clone the FinniGan Repo using ‘git clone’
- In the Directory ‘data’ create an empty directory ‘output’, all the videos to be processed and all the datapoints generated will be saved in the ‘Videos’ and ‘output’ directory respectively.
- (Make sure to be in FinniGAN directory befire this) Run the ‘FrameExt.py’ script it uses OpenCV’s VidCapture method to extract the frame and save them in the output directory.
- If training using the BCE loss,no further Changes to be made, else remove the ‘nn.Sigmoid’ activation from the Discriminator from ‘model.py’.
- Finally run the ‘test.py’ to train the model.
-
The Pre-Trained model can be used from the ‘logs’ directory, to test on your images:
- Pass 2 consecutive frame tensors, stacked (each pixel value being the avg of two) upon each other having shape (1,3,256,256), the output you receive will be the middle generated frame.
- You can also use the showImgTEST() method on a predefine dataset to test the results.
-
This paper can also be referenced: FREGAN
-
Notifications
You must be signed in to change notification settings - Fork 1
zfyre/FiNNiGAN-Implementation
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
Implemented The Frame-Interpolation Using GANs
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published