Dataset Link: Click
Download dataset.zip file from above link , extract and add it to the same folder.
For detailed instruction about training the model see train.ipynb file.
To test the model on two test videos run detection.py file.
python3 detection.py
Sample Output frame :
