问题标签 [yolov5]

我只想检测指定区域中的对象并忽略 ROI 之外的所有其他检测。

我有一个看起来像笔的单个对象的数据集，我需要对它们进行计数。我想使用对象检测和对象跟踪器。经过大量研究，我决定使用yolov5和深度排序。在标记数据集并对其进行训练之后，它工作得很好，直到出现大问题。当“笔”以对角线方式并排堆叠时，Yolo 无法检测到它们，因为标签彼此太靠近。我试图将阈值保持提高到 0.95，但仍然存在问题。我意识到在我的数据集中，大多数标签至少有 60% 的重叠，这意味着在大多数情况下，标签包含一个对象和一半或更多。所以最大的问题是，在 Yolo 不具备对角标注能力的情况下，如何在如此拥挤的场景中检测物体？

简要说明

最近开始学习Object Detection，刚开始学习PyTorch，YOLOv5。所以我想为什么不建一个小项目来学习呢？用它来训练检测皮卡丘。

问题

我已经成功地用皮卡丘训练了模型，然后使用训练过的权重和我自己编写的 Python 脚本/代码来使用测试图像检测皮卡丘，现在，问题来了，皮卡丘可以成功检测到，但所有结果都显示为蓝色变色，这应该怎么办变成黄色，全部变成蓝色，蓝色变成黄色。

图 1 结果图像显示为蓝色（少数示例输出）

附加信息

我已经将这个项目推送到了 GitHub，请随意下载或拉取它进行调试。

包含所有文件的 GitHub 存储库

任何解决方案/建议都会有所帮助，谢谢。

编码

我试图从这个站点将 YOLO v5 格式转换为 50one 格式。

原始代码如下。

但是，我收到以下错误：

我不知道是什么导致了这个错误，任何指导将不胜感激！

我正在尝试在自定义数据集上训练 yolov5，但出现此错误。

/home/cardamom/.local/lib/python3.8/site-packages/torch/nn/functional.py:718：UserWarning：命名张量及其所有相关 API 是实验性功能，可能会发生变化。在它们稳定发布之前，请不要将它们用于任何重要的事情。（在/pytorch/c10/core/TensorImpl.h:1156内部触发。） return torch.max_pool2d(input, kernel_size, stride, padding, dilation, ceil_mode) 模型总结：165层，6.8665e+06个参数，6.8665e+ 06 个梯度，12.1 GFLOPS

优化器组：54 个 .bias，60 个 conv.weight，51 个其他缓存标签 /home/cardamom/Desktop/yolov5-master/data/train_kinoo.txt（找到 21 个，丢失 0 个，空的，0 个重复的 Traceback（最近一次调用最后一次） ): 文件“train.py”，第 404 行，在 train(hyp) 文件“train.py”，第 165 行，在 train assert mlc < nc, '标签类 %g 在 %s 中超过 nc=%g。更正你的标签或您的模型。' % (mlc, nc, opt.cfg) AssertionError：标签类 33 超过 ././models/yolov5s.yaml 中的 nc=3。更正您的标签或模型。

我查找了此错误的原因。我按照这个线程自定义数据训练问题来解决这个问题。我确保标签只有三个类，并且 yolov5s.yaml 中的类数有 nc = 3。我还确保 dataset.yaml 也有 nc = 3。我不知道它在哪里选择 nc = 33在标签中。

帮我！在过去的三天里，我一直在这个问题上。

我在 yoloV5s 模型中训练了我的自定义数据集，我的推理准确率达到了 80%。现在我需要通过添加更多图像和标签来提高准确性。

我的问题是，我已经训练了 10,000 多个标签以达到 80%，我花了 7 个小时。我是否需要将旧的 10,000 多个数据包含在我的新数据中，而新数据只有 1000 来训练和提高我的准确性？

即使我添加了一个新类，有什么方法可以只包含新数据来重新训练模型？

如何节省时间和空间？

有人可以帮我将 YOLOv5 PyTorch 模型转换为 ONNX 或 TensorFlow 格式，以便能够将其与 OpenCV C++ 推理一起使用吗？

我使用本教程使用 colab 训练模型：https ://blog.roboflow.com/how-to-train-yolov5-on-a-custom-dataset/

也许在 colab 中有一种转换方式，或者我应该使用任何其他工具？

我正在使用以下内容来实时从打开的应用程序中获取屏幕截图。如何运行仅检测来自抓取屏幕的输入的 detect.py？谢谢。

我的抓屏

YoloV5 检测.py

现在它只检测图像而不是我的实时抓取屏幕

预期结果：

正在下载https://github.com/ultralytics/yolov5/releases/download/v5.0/yolov5s.pt到 /root/.cache/torch/hub/ultralytics_yolov5_master/yolov5s.pt... 100%|#### ##################################| 14.1M/14.1M [00:05<00:00, 2.72MB/s]

融合层...模型摘要：224 层，7266973 个参数，0 个渐变添加自选图形...地址处没有 I2C 设备：0x69 回溯（最后一次调用）：文件“maskcamera.py”，第 224 行，在 identify_maskman(0 ) 文件“maskcamera.py”，第 185 行，在 identify_maskman 预测 = model.predict(data) 文件“/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py”，行948，在getattr 类型（自我）中。name , name)) AttributeError: 'AutoShape' object has no attribute 'predict' Exiting... 清理引脚

我使用 php 在 yolov5 上工作，但遇到了一个问题。我通过windows PowerShell运行yolov5/train.py文件，没有问题。

python train.py --img 640 --batch 8 --epochs 3 --data ../datasets/777/777.yaml --workers 0 --image-weights ''

有用。但是当我用 cmd 或 php 运行相同的代码时，我得到了这个错误：