基于flask的服务器后端
git clone https://github.com/cqbu/driving_detection_server.git
conda create -n driving python=3.8 -y
conda activate driving
# 安装PyTorch(也可通过pip安装):请确保cudatoolkit版本与系统cuda版本一致
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3 -c pytorch -c conda-forge
pip install -r requirements.txt
# 编译CLRNet
cd CLRNet
python setup.py build develop
python app.py
- 本仓库使用的前端:driving_detection_app
- PyTorch安装
- yolov5
- CLRNet