pineappleSoup's picture
Update README.md
dff680f verified
|
raw
history blame
666 Bytes

Stroke Interpolation Model

Installation

pip install opencv-python 
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128

Run the Program

py combined.py

This will load the model from checkpoints.

Evaluate

py generate_eval.py

This generates images for evaluation.

py eval.py

This evaluates the generated images.

Test

py test.py

Add frames in test_frames folder.

py video.py

This combines those 3 frames into .mp4 format.

Dataset

Dataset is available at: Google Drive Link