This project was created as a part of master's thesis at the FEE CTU.
The project is implemented using Python 3.8.5 and with PyTorch 1.8.1. Other used libraries can be found in the requirements.txt file. The original code accompanying the Zhang et al. can be found on Microsoft's GitHub repository.
First of all Python version at least 3.8 must be installed. Then install pip, which is required to install the project's dependencies. Run following command to install the dependencies:
pip install -r requirements.txtThe main application entrypoint main.py supports 4 basic commands:
- train
- inference
- validation
- evaluation
The application is written using Click library, so every command has automatically generated help pages. Each can be invoked by running:
python main.py <command> --help}.Examples of running each command are in file example.sh.
Additionally, the project offers various visualizations of predicted frames. The visualizations are invoked same way as the main entrypoint and are located in file visualization.py:
python visualization.py <command> --help}.Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.
Created by Hynek Davídek.