基于yolov5的区域目标检测
先上图片
使用的是yolov5官方的yolov5s.pt权重,可以根据代码替换为自己的权重或者其他yolov5官方权重,代码的github链接:
https://github.com/Kinseys/changeable_region_detection
首先是装环境,用的pycharm+conda的话在terminal里面输入:
pip install -r requirements.txt
装好环境以后可以将自己想要检测的图片放入data/images/文件里面:然后就可以直接运行detect.py
python detect.py
如果想要运行自己的摄像头:
python detect.py --source 0
或者在代码中将256 行 default后面内容改为 0 :
pa