如何使 YOLO 输出到另一个程序的输入？

【问题标题】：How to make YOLO output to the input for another program?如何使 YOLO 输出到另一个程序的输入？
【发布时间】：2021-11-20 01:37:00
【问题描述】：

我是这个领域的新手。我正在使用 YOLOv5 进行自定义对象检测。它已成功实施。现在，我还想将 YOLOv5 的输出用作另一个输入。实际上，我想制作一个程序，如果该人没有戴上口罩，它会戴上口罩说话。这些类是：“with_mask”和“without_mask”。我不知道该怎么做。请帮我解决这个问题。

【问题讨论】：

请编辑问题以将其限制为具有足够详细信息的特定问题，以确定适当的答案。

标签： python pytorch object-detection yolo yolov5

【解决方案1】：

我不知道你是如何实现YOLOv5的，所以你必须自己检查pred的部分......

from gtts import gTTS
import os
  
mytext = 'Pls wear mask'
language = 'en'
myobj = gTTS(text=mytext, lang=language, slow=False)
  
myobj.save("mask.mp3")
 

if pred["label"] == "without_mask":
   os.system("mpg321 mask.mp3")

【讨论】：

感谢您的回复。但是，我不太确定应该如何以及在哪里使用您提供的代码。我使用了 ultralytics 提供的 YOLOv5 模型，并针对自定义数据集对其进行了训练。我没有做任何其他更改。那么，我应该在 detect.py 文件的哪个部分使用此代码？提前谢谢你