-3

我最近对这个关于图像分类的机器学习主题产生了兴趣。
我绝不是程序员,但我是一个对此非常感兴趣的农民,而检测水果和蔬菜的质量是一项非常繁琐且耗时的任务,特别是如果您没有钱购买工业机械以中小型规模执行此任务。

我最近遇到了这个教程(不得不修复很多错误,因为它写得非常糟糕,但它确实有效):

https://imaginghub.com/projects/148-how-to-distinguish-apples-and-pears-with-raspberry-pi/documentation

这基本上是未来水果/蔬菜质量分级机的基石。

这条传送带将有一个 Arduino,它将接收来自 Python 程序的输出,该输出应激活伺服系统以将每个水果/蔬菜重定向到自己的篮子。

现在我想知道如何从网络获取标签输出并将其转换为数字,例如:

苹果 = 1,橙子 = 2,黄瓜 = 3...

因此,只要是苹果,Arduino 就会收到一个 1 来点亮一个 LED(首先是这个,然后是伺服),橙色和黄瓜等也会发生同样的情况。

这是我认为必须对框标签输出 Deploy.py 和 yolo_net.py 做一些事情的 2 个代码

Deploy.py 是我为获取实时摄像头检测器而运行的文件:

部署.ipynb

这是 yolo_net.py (如果它有助于以某种方式回答我的问题):

yolo_net.py

4

1 回答 1

0

您所要做的就是修改yolo_net.py文件并转到def draw_detections(self, image, boxes):第 63 行,它应该说

cv2.putText  (image, class_name, (left, top - 12), 0, 1e-3 * h, self.colors[class_indx], thick//2)  

在该行下,您应该添加您希望 arduino 执行的操作。

于 2017-11-22T00:13:57.930 回答