Skip to content

Update Installation Process #13

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
Metablockofwood wants to merge 8 commits into
base: master
Choose a base branch
from
Open

Update Installation Process #13

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
6681653
Adding a requirements file.
Metablockofwood Jan 13, 2021
7790b77
Updating the readme to include the usage of the newly added requireme…
Metablockofwood Jan 13, 2021
87aaba1
Removing dollar signs to conform to how commands are currently listed…
Metablockofwood Jan 13, 2021
29a5f31
Torch 1.7.0 supports the following compute capibilities ['sm_37', 'sm…
Metablockofwood Jan 13, 2021
a0f9714
Try to add new lines back in.
Metablockofwood Jan 13, 2021
bd512ea
Undoing changes from last commit.
Metablockofwood Jan 14, 2021
e99aa72
Formatting. Adding colons to read me.
Metablockofwood Jan 14, 2021
08b582d
Add back in compute capability 6.1 . PyTorch 1.7.0 compute capability…
Metablockofwood Jan 22, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 5 additions & 5 deletions PWCNet/correlation_package_pytorch1_0/setup.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -15,16 +15,16 @@
#'-gencode', 'arch=compute_35,code=sm_35',
'-gencode', 'arch=compute_37,code=sm_37',
'-gencode', 'arch=compute_50,code=sm_50',
'-gencode', 'arch=compute_52,code=sm_52',
'-gencode', 'arch=compute_53,code=sm_53',
#'-gencode', 'arch=compute_52,code=sm_52',
#'-gencode', 'arch=compute_53,code=sm_53',
'-gencode', 'arch=compute_60,code=sm_60',
'-gencode', 'arch=compute_61,code=sm_61',
'-gencode', 'arch=compute_62,code=sm_62',
#'-gencode', 'arch=compute_62,code=sm_62',
'-gencode', 'arch=compute_70,code=sm_70',
'-gencode', 'arch=compute_72,code=sm_72',
#'-gencode', 'arch=compute_72,code=sm_72',
'-gencode', 'arch=compute_75,code=sm_75',
'-gencode', 'arch=compute_80,code=sm_80',
'-gencode', 'arch=compute_86,code=sm_86',
#'-gencode', 'arch=compute_86,code=sm_86',
#'-gencode', 'arch=compute_75,code=sm_75',
#'-gencode', 'arch=compute_75,code=compute_75'
]
Expand Down
37 changes: 20 additions & 17 deletions README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ This is the source code for the video interpolation application **Dain-App**, de
### Table of Contents
1. [Introduction](#introduction)
1. [Citation](#citation)
1. [Requirements and Dependencies](#requirements-and-dependencies)
1. [Prerequisites](#prerequisites)
1. [Installation](#installation)
1. [Running application with interface](#running-application-with-interface)
1. [Running application with command line](#running-application-with-command-line)
Expand Down Expand Up @@ -48,25 +48,28 @@ If you find the code and datasets useful in your research, please cite:
year={2018}
}

### Requirements and Dependencies
- numba=0.51.2
- numpy=1.19.2
- opencv-python=4.4.0.46
- pillow=8.0.1
- pyqt5=5.15.1
- python=3.8.5
- scikit-learn=0.23.2
- scipy=1.5.4
- torch=1.7.0+cu110
- torchvision=0.8.1+cu110
- tqdm=4.51.0
### Prerequisites
- Python 3.8.5
- CUDA 11
- ffmpeg

### Installation
Check out the Colab code:
Check out the example Colab code:
[<img src="https://colab.research.google.com/assets/colab-badge.svg" align="center">](https://colab.research.google.com/github/BurguerJohn/Dain-App/blob/master/Dain_App_Colab.ipynb)

Remember you need to build the .cuda scripts before the app can work.
Install the requirements:

pip install -r requirements.txt -f https://download.pytorch.org/whl/torch_stable.html

Build the CUDA packages:

cd my_package
./build.sh

cd ../PWCNet/correlation_package_pytorch1_0
./build.sh

cd ../../

### Running application with interface
python my_design.py
Expand All @@ -80,7 +83,7 @@ A example of a working code:
python my_design.py -cli --input "gif/example.gif" -o "example_folder/" -on "interpolated.gif" -m "model_weights/best.pth" -fh 3 --interpolations 2 --depth_awarenes 0 --loop 0 -p 0 --alpha 0 --check_scene_change 10 --png_compress 0 --crf 1 --pixel_upscale_downscale_before 1 --pixel_downscale_upscale_after 1 --pixel_upscale_after 1 --mute_ffmpeg 1 --split_size_x -1 --split_size_y -1 --split_pad 150 --half 0 --step_extract 1 --step_interpolate 1 --batch_size 1 --use_benchmark 0 --force_flow 1 --smooth_flow 0 --downscale -1 --fast_mode 0

### Training New Models
Currently Dain-App training code is broken, to train new models, use the DAIN github and import the models to Dain-App
Currently Dain-App training code is broken, to train new models, use the [DAIN](https://github.com/baowenbo/DAIN) github and import the models to Dain-App

### Google Colab Demo
[<img src="https://colab.research.google.com/assets/colab-badge.svg" align="center">](https://colab.research.google.com/github/BurguerJohn/Dain-App/blob/master/Dain_App_Colab.ipynb)
Expand All @@ -89,4 +92,4 @@ Currently Dain-App training code is broken, to train new models, use the DAIN gi
[Gabriel Poetsch](mailto:griskai.yt@gmail.com)

### License
See [MIT License](https://github.com/BurguerJohn/Dain-App/blob/master/LICENSE)
See [MIT License](https://github.com/BurguerJohn/Dain-App/blob/master/LICENSE)
14 changes: 7 additions & 7 deletions my_package/Arch.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,17 +18,17 @@ def AllArch():
#'-gencode', 'arch=compute_30,code=sm_30',
#'-gencode', 'arch=compute_32,code=sm_32',
#'-gencode', 'arch=compute_35,code=sm_35',
#'-gencode', 'arch=compute_37,code=sm_37',
'-gencode', 'arch=compute_37,code=sm_37',
'-gencode', 'arch=compute_50,code=sm_50',
'-gencode', 'arch=compute_52,code=sm_52',
'-gencode', 'arch=compute_53,code=sm_53',
#'-gencode', 'arch=compute_52,code=sm_52',
#'-gencode', 'arch=compute_53,code=sm_53',
'-gencode', 'arch=compute_60,code=sm_60',
'-gencode', 'arch=compute_61,code=sm_61',
'-gencode', 'arch=compute_62,code=sm_62',
#'-gencode', 'arch=compute_62,code=sm_62',
'-gencode', 'arch=compute_70,code=sm_70',
'-gencode', 'arch=compute_72,code=sm_72',
#'-gencode', 'arch=compute_72,code=sm_72',
'-gencode', 'arch=compute_75,code=sm_75',
#'-gencode', 'arch=compute_80,code=sm_80',
'-gencode', 'arch=compute_80,code=sm_80',
#'-gencode', 'arch=compute_86,code=sm_86',
#'-gencode', 'arch=compute_86,code=compute_86'
]
Expand All @@ -38,4 +38,4 @@ def GetArchs():
if getAll:
return AllArch()
else:
return MyArch()
return MyArch()
14 changes: 7 additions & 7 deletions my_package/DepthFlowProjection/Arch.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,18 +18,18 @@ def AllArch():
#'-gencode', 'arch=compute_30,code=sm_30',
#'-gencode', 'arch=compute_32,code=sm_32',
#'-gencode', 'arch=compute_35,code=sm_35',
#'-gencode', 'arch=compute_37,code=sm_37',
'-gencode', 'arch=compute_37,code=sm_37',
'-gencode', 'arch=compute_50,code=sm_50',
'-gencode', 'arch=compute_52,code=sm_52',
'-gencode', 'arch=compute_53,code=sm_53',
#'-gencode', 'arch=compute_52,code=sm_52',
#'-gencode', 'arch=compute_53,code=sm_53',
'-gencode', 'arch=compute_60,code=sm_60',
'-gencode', 'arch=compute_61,code=sm_61',
'-gencode', 'arch=compute_62,code=sm_62',
#'-gencode', 'arch=compute_62,code=sm_62',
'-gencode', 'arch=compute_70,code=sm_70',
'-gencode', 'arch=compute_72,code=sm_72',
#'-gencode', 'arch=compute_72,code=sm_72',
'-gencode', 'arch=compute_75,code=sm_75',
'-gencode', 'arch=compute_80,code=sm_80',
'-gencode', 'arch=compute_86,code=sm_86',
#'-gencode', 'arch=compute_86,code=sm_86',
#'-gencode', 'arch=compute_86,code=compute_86'
]

Expand All @@ -38,4 +38,4 @@ def GetArchs():
if getAll:
return AllArch()
else:
return MyArch()
return MyArch()
14 changes: 7 additions & 7 deletions my_package/FilterInterpolation/Arch.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,18 +18,18 @@ def AllArch():
#'-gencode', 'arch=compute_30,code=sm_30',
#'-gencode', 'arch=compute_32,code=sm_32',
#'-gencode', 'arch=compute_35,code=sm_35',
#'-gencode', 'arch=compute_37,code=sm_37',
'-gencode', 'arch=compute_37,code=sm_37',
'-gencode', 'arch=compute_50,code=sm_50',
'-gencode', 'arch=compute_52,code=sm_52',
'-gencode', 'arch=compute_53,code=sm_53',
#'-gencode', 'arch=compute_52,code=sm_52',
#'-gencode', 'arch=compute_53,code=sm_53',
'-gencode', 'arch=compute_60,code=sm_60',
'-gencode', 'arch=compute_61,code=sm_61',
'-gencode', 'arch=compute_62,code=sm_62',
#'-gencode', 'arch=compute_62,code=sm_62',
'-gencode', 'arch=compute_70,code=sm_70',
'-gencode', 'arch=compute_72,code=sm_72',
#'-gencode', 'arch=compute_72,code=sm_72',
'-gencode', 'arch=compute_75,code=sm_75',
'-gencode', 'arch=compute_80,code=sm_80',
'-gencode', 'arch=compute_86,code=sm_86',
#'-gencode', 'arch=compute_86,code=sm_86',
#'-gencode', 'arch=compute_86,code=compute_86'
]

Expand All @@ -38,4 +38,4 @@ def GetArchs():
if getAll:
return AllArch()
else:
return MyArch()
return MyArch()
14 changes: 7 additions & 7 deletions my_package/FlowProjection/Arch.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,18 +18,18 @@ def AllArch():
#'-gencode', 'arch=compute_30,code=sm_30',
#'-gencode', 'arch=compute_32,code=sm_32',
#'-gencode', 'arch=compute_35,code=sm_35',
#'-gencode', 'arch=compute_37,code=sm_37',
'-gencode', 'arch=compute_37,code=sm_37',
'-gencode', 'arch=compute_50,code=sm_50',
'-gencode', 'arch=compute_52,code=sm_52',
'-gencode', 'arch=compute_53,code=sm_53',
#'-gencode', 'arch=compute_52,code=sm_52',
#'-gencode', 'arch=compute_53,code=sm_53',
'-gencode', 'arch=compute_60,code=sm_60',
'-gencode', 'arch=compute_61,code=sm_61',
'-gencode', 'arch=compute_62,code=sm_62',
#'-gencode', 'arch=compute_62,code=sm_62',
'-gencode', 'arch=compute_70,code=sm_70',
'-gencode', 'arch=compute_72,code=sm_72',
#'-gencode', 'arch=compute_72,code=sm_72',
'-gencode', 'arch=compute_75,code=sm_75',
'-gencode', 'arch=compute_80,code=sm_80',
'-gencode', 'arch=compute_86,code=sm_86',
#'-gencode', 'arch=compute_86,code=sm_86',
#'-gencode', 'arch=compute_86,code=compute_86'
]

Expand All @@ -38,4 +38,4 @@ def GetArchs():
if getAll:
return AllArch()
else:
return MyArch()
return MyArch()
14 changes: 7 additions & 7 deletions my_package/Interpolation/Arch.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,18 +18,18 @@ def AllArch():
#'-gencode', 'arch=compute_30,code=sm_30',
#'-gencode', 'arch=compute_32,code=sm_32',
#'-gencode', 'arch=compute_35,code=sm_35',
#'-gencode', 'arch=compute_37,code=sm_37',
'-gencode', 'arch=compute_37,code=sm_37',
'-gencode', 'arch=compute_50,code=sm_50',
'-gencode', 'arch=compute_52,code=sm_52',
'-gencode', 'arch=compute_53,code=sm_53',
#'-gencode', 'arch=compute_52,code=sm_52',
#'-gencode', 'arch=compute_53,code=sm_53',
'-gencode', 'arch=compute_60,code=sm_60',
'-gencode', 'arch=compute_61,code=sm_61',
'-gencode', 'arch=compute_62,code=sm_62',
#'-gencode', 'arch=compute_62,code=sm_62',
'-gencode', 'arch=compute_70,code=sm_70',
'-gencode', 'arch=compute_72,code=sm_72',
#'-gencode', 'arch=compute_72,code=sm_72',
'-gencode', 'arch=compute_75,code=sm_75',
'-gencode', 'arch=compute_80,code=sm_80',
'-gencode', 'arch=compute_86,code=sm_86',
#'-gencode', 'arch=compute_86,code=sm_86',
#'-gencode', 'arch=compute_86,code=compute_86'
]

Expand All @@ -38,4 +38,4 @@ def GetArchs():
if getAll:
return AllArch()
else:
return MyArch()
return MyArch()
14 changes: 7 additions & 7 deletions my_package/InterpolationCh/Arch.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,18 +18,18 @@ def AllArch():
#'-gencode', 'arch=compute_30,code=sm_30',
#'-gencode', 'arch=compute_32,code=sm_32',
#'-gencode', 'arch=compute_35,code=sm_35',
#'-gencode', 'arch=compute_37,code=sm_37',
'-gencode', 'arch=compute_37,code=sm_37',
'-gencode', 'arch=compute_50,code=sm_50',
'-gencode', 'arch=compute_52,code=sm_52',
'-gencode', 'arch=compute_53,code=sm_53',
#'-gencode', 'arch=compute_52,code=sm_52',
#'-gencode', 'arch=compute_53,code=sm_53',
'-gencode', 'arch=compute_60,code=sm_60',
'-gencode', 'arch=compute_61,code=sm_61',
'-gencode', 'arch=compute_62,code=sm_62',
#'-gencode', 'arch=compute_62,code=sm_62',
'-gencode', 'arch=compute_70,code=sm_70',
'-gencode', 'arch=compute_72,code=sm_72',
#'-gencode', 'arch=compute_72,code=sm_72',
'-gencode', 'arch=compute_75,code=sm_75',
'-gencode', 'arch=compute_80,code=sm_80',
'-gencode', 'arch=compute_86,code=sm_86',
#'-gencode', 'arch=compute_86,code=sm_86',
#'-gencode', 'arch=compute_86,code=compute_86'
]

Expand All @@ -38,4 +38,4 @@ def GetArchs():
if getAll:
return AllArch()
else:
return MyArch()
return MyArch()
14 changes: 7 additions & 7 deletions my_package/MinDepthFlowProjection/Arch.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,18 +18,18 @@ def AllArch():
#'-gencode', 'arch=compute_30,code=sm_30',
#'-gencode', 'arch=compute_32,code=sm_32',
#'-gencode', 'arch=compute_35,code=sm_35',
#'-gencode', 'arch=compute_37,code=sm_37',
'-gencode', 'arch=compute_37,code=sm_37',
'-gencode', 'arch=compute_50,code=sm_50',
'-gencode', 'arch=compute_52,code=sm_52',
'-gencode', 'arch=compute_53,code=sm_53',
#'-gencode', 'arch=compute_52,code=sm_52',
#'-gencode', 'arch=compute_53,code=sm_53',
'-gencode', 'arch=compute_60,code=sm_60',
'-gencode', 'arch=compute_61,code=sm_61',
'-gencode', 'arch=compute_62,code=sm_62',
#'-gencode', 'arch=compute_62,code=sm_62',
'-gencode', 'arch=compute_70,code=sm_70',
'-gencode', 'arch=compute_72,code=sm_72',
#'-gencode', 'arch=compute_72,code=sm_72',
'-gencode', 'arch=compute_75,code=sm_75',
'-gencode', 'arch=compute_80,code=sm_80',
'-gencode', 'arch=compute_86,code=sm_86',
#'-gencode', 'arch=compute_86,code=sm_86',
#'-gencode', 'arch=compute_86,code=compute_86'
]

Expand All @@ -38,4 +38,4 @@ def GetArchs():
if getAll:
return AllArch()
else:
return MyArch()
return MyArch()
14 changes: 7 additions & 7 deletions my_package/SeparableConv/Arch.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,18 +18,18 @@ def AllArch():
#'-gencode', 'arch=compute_30,code=sm_30',
#'-gencode', 'arch=compute_32,code=sm_32',
#'-gencode', 'arch=compute_35,code=sm_35',
#'-gencode', 'arch=compute_37,code=sm_37',
'-gencode', 'arch=compute_37,code=sm_37',
'-gencode', 'arch=compute_50,code=sm_50',
'-gencode', 'arch=compute_52,code=sm_52',
'-gencode', 'arch=compute_53,code=sm_53',
#'-gencode', 'arch=compute_52,code=sm_52',
#'-gencode', 'arch=compute_53,code=sm_53',
'-gencode', 'arch=compute_60,code=sm_60',
'-gencode', 'arch=compute_61,code=sm_61',
'-gencode', 'arch=compute_62,code=sm_62',
#'-gencode', 'arch=compute_62,code=sm_62',
'-gencode', 'arch=compute_70,code=sm_70',
'-gencode', 'arch=compute_72,code=sm_72',
#'-gencode', 'arch=compute_72,code=sm_72',
'-gencode', 'arch=compute_75,code=sm_75',
'-gencode', 'arch=compute_80,code=sm_80',
'-gencode', 'arch=compute_86,code=sm_86',
#'-gencode', 'arch=compute_86,code=sm_86',
#'-gencode', 'arch=compute_86,code=compute_86'
]

Expand All @@ -38,4 +38,4 @@ def GetArchs():
if getAll:
return AllArch()
else:
return MyArch()
return MyArch()
14 changes: 7 additions & 7 deletions my_package/SeparableConvFlow/Arch.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,18 +18,18 @@ def AllArch():
#'-gencode', 'arch=compute_30,code=sm_30',
#'-gencode', 'arch=compute_32,code=sm_32',
#'-gencode', 'arch=compute_35,code=sm_35',
#'-gencode', 'arch=compute_37,code=sm_37',
'-gencode', 'arch=compute_37,code=sm_37',
'-gencode', 'arch=compute_50,code=sm_50',
'-gencode', 'arch=compute_52,code=sm_52',
'-gencode', 'arch=compute_53,code=sm_53',
#'-gencode', 'arch=compute_52,code=sm_52',
#'-gencode', 'arch=compute_53,code=sm_53',
'-gencode', 'arch=compute_60,code=sm_60',
'-gencode', 'arch=compute_61,code=sm_61',
'-gencode', 'arch=compute_62,code=sm_62',
#'-gencode', 'arch=compute_62,code=sm_62',
'-gencode', 'arch=compute_70,code=sm_70',
'-gencode', 'arch=compute_72,code=sm_72',
#'-gencode', 'arch=compute_72,code=sm_72',
'-gencode', 'arch=compute_75,code=sm_75',
'-gencode', 'arch=compute_80,code=sm_80',
'-gencode', 'arch=compute_86,code=sm_86',
#'-gencode', 'arch=compute_86,code=sm_86',
#'-gencode', 'arch=compute_86,code=compute_86'
]

Expand All @@ -38,4 +38,4 @@ def GetArchs():
if getAll:
return AllArch()
else:
return MyArch()
return MyArch()
10 changes: 10 additions & 0 deletions requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
numba==0.51.2
numpy==1.19.2
opencv-python-headless==4.4.0.46
Pillow==8.0.1
PyQt5==5.15.1
scikit-learn==0.23.2
scipy==1.5.4
torch==1.7.0+cu110
torchvision==0.8.1+cu110
tqdm==4.51.0