Describe the bug
Running video.py with the provided model ('face_mask.ckpt') with any video, fails to load the model state_dict with KeyError: 'state_dict'
To Reproduce
Steps to reproduce the behavior:
Download repo and create a virtualenv(in PyCharm) from requirements.txt.
PyTorch, Pandas and PyTorch Lightning do not install out of the box, they need to be installed manually because of version errors.
Expected behavior
Output a video with box overlays over faces with or without masks.
Screenshots
Desktop (please complete the following information):
- OS: Windows 10
- Python Version 3.7.7
- PyTorch 1.5 last stable version