从YOLOv3中提取检测列表

问题描述 投票:0回答:1

[我正在使用YOLOv3开发对象检测应用程序,我对模型进行了重新训练以检测两个类,感谢Repo的创建者,到目前为止,它已经成功完成了。现在,我必须将模型部署在jetson nano套件上,它应该连续运行并提取指定类别,置信度和边界框坐标的检测列表,然后我应该对前面提到的参数进行一些处理,然后使用任何通信协议将其发送到无论如何,另一个ECU如何从YOLOv3中提取上述参数并使该代码连续运行

artificial-intelligence embedded-linux object-detection yolo darknet
1个回答
0
投票

如果您使用Python脚本进行YOLOv3检测,您将在脚本中遇到此部分:

detections = darknet.detect_image(netMain, metaMain, darknet_image, thresh=0.25)
image = cvDrawBoxes(detections, frame_resized)
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)

这里的检测应该是字典或某种数组(在不同的实现中会有所不同)。它应该包含detected_classesnumber_of_detectionsconfidence_scoresbounding_boxes。您可以修改代码以提取所需的任何内容。上面的代码来自this存储库。我还使用YOLOv3进行连续图像检测。假设可以判断图片中是否包含人物。

[注意:如果我在YOLOv3的C ++脚本中没有记错的话,应该有一个与我上面提到的相似的部分。

© www.soinside.com 2019 - 2024. All rights reserved.