Skip to content

Latest commit

 

History

History
 
 

RAFT

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
core
core
 
 
README.md
README.md
 
 
dataset_conv.py
dataset_conv.py
 
 
raft-model.pth
raft-model.pth
 
 
raft_optical_flow.py
raft_optical_flow.py
 
 
requirements-raft.txt
requirements-raft.txt
 
 

README.md

Optical flow with RAFT

Dataset tranformation using Optical Flow RAFT method

Requirements

You can find all requirements in requirements-raft.txt file

Required Data

To convert your dataset to Optical Flow, you will need to create directory strucutre as below:

├── dataset
│   ├── Fight
│   │   ├── 0
│   │   │   ├── frame0.png
│   │   │   ├── frame1.png
│   │   │   ... 
│   │   └── 1
│   │   │   ├── frame0.png
│   │   │   ├── frame1.png
│   │       ... 
│   └── NonFight
│   │   ├── 0
│   │   │   ├── frame0.png
│   │   │   ├── frame1.png
│   │   │   ... 
│   │   └── 1
│   │   │   ├── frame0.png
│   │   │   ├── frame1.png
│   │       ...

Using

python dataset_conv.py --model raft-model.pth --path path//to//dataset// --path_save path//to//save//dir//