在项目中遇到一个需求,需要对图片中的箭头方向进行判断,本来是使用YOLOv8算法来实现的,但是发现YOLO的效果对箭头的识别效果很差,不管是分类算法还是检测算法,效果都不理想,因此试一试通过openCV对箭头方向进行判断,发现效果还可以。
下面附上完整的代码和原理。
一、运行环境
运行环境是windows11+python(3.12)+opencv,
运行环境的完整配置如下所示:
certifi==2024.12.14
charset-normalizer==3.4.1
colorama==0.4