- Either Linux or Windows. We recommend Linux for better performance.
- Python 3.6+ and PyTorch 1.7+.
To install dependencies run this command:
pip install -r requirements.txt
- download model from this link: https://drive.google.com/file/d/1W6qZeutnqnp3YX9w4iYgR44xsoi_64ff/view?usp=sharing
- and put them on weights directory
To run inference on an image or video Run this command:
python main.py
- The data set used is here : https://t.ly/NZWj