Skip to content

Latest commit

 

History

History
13 lines (7 loc) · 315 Bytes

README.md

File metadata and controls

13 lines (7 loc) · 315 Bytes

Prerequisites

Python 3.8
pip install torch==1.8.2+cu111 torchvision==0.9.2+cu111 torchaudio==0.8.2 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html
pip install opencv-python==4.5.5.64 opencv-contrib-python==4.5.5.64
pip install -r requirements.txt

Train

python3 main.py