基于volov5实现目标检测后输出目标位置

基于volov5实现目标检测后输出目标位置
找了好久教程都找不到,怎么输出目标位置,找了好多天,自己摸索的
直接在detect.py文件中找到save_one_box在下面粘贴这些代码就可以了,不需要改动其他文件,完美解决
基于volov5实现目标检测后输出目标位置_第1张图片
代码如下:

x1 = int(xyxy[0].item())
y1 = int(xyxy[1].item())
x2 = int(xyxy[2].item())
y2 = int(xyxy[3].item())
class_index = cls
object_name = names[int(cls)]
print('位置: ', x1, y1, x2, y2)
print('属性ID ', class_index)
print('属性 ', object_name)
by:我是小杰啊

完成功能的截图如下:
基于volov5实现目标检测后输出目标位置_第2张图片

你可能感兴趣的:(Yolov5,python,人工智能)