目标:利用yolov5和realsense摄像头做一个简单的人类检测。
参考教程目标检测---教你利用yolov5训练自己的目标检测模型_目标检测 教你利用-优快云博客一步一步操作。
1.项目克隆和环境配置
首先下载好yolov5.5分支的源码,但是在环境配置时踩了很多的坑。
首先在用conda创建虚拟环境的时候注意不要环境嵌套,先退出上一个环境再创建,使用conda info指令查看shell level应该为1,大于1则说明发生了嵌套。
还有一开始我的虚拟环境中找到的pip和python都是全局的,为避免后续出各种问题,我又删掉重新创建了一个虚拟环境,得到以下输出就没问题可以进行下一步了:
这里记录一下我使用的环境:
Python 3.9.19
CUDA 11.7
cuDNN 8.6.0
pytorch 2.3.1
numpy 1.23.0