《做个 ROS 2 视觉检测开源库-使用 Python 调用 YOLO》 YOLOv5 使用pytorch 实现,所以我们肯定要用python来编写代码,在 python 中,可以将 YOLOv5 当作第三方库一样导入使用,在 chapt9/learn_yolo 目录下新建 detect.py ,接着编写如下代码: import cv2 # 用于图像处理 import yolov5 # 导入YOLOv5模块 # 读取...
为了加载和显示图片,我们首先在代码中导入了 cv2 模块,接着导入了 yolov5 库。导入完成后,就开始读取图像,接着将图像从 BGR 颜色空间转换为 RGB 颜色空间,以便 YOLOv5 检测使用。然后使用 yolov5 加载预训练模型 ,接着就可以调用模型对图像进行检测,检测完成后首先获取结果张量,然后从其中获取检测到物体的位置数组...