0

我正在处理带有图表的文本图像。我的图像基本上是黑白的，我不明白为什么我想要图像中的颜色。我在默认设置下得到了一些不错的结果，但我也想在灰度图像上进行测试。我使用本教程作为基础，默认情况下使用AlexyAB'srepo for darknet. 我想我必须将config文件更改为：

channels=3 # I think I have to change it to 0
momentum=0.9
decay=0.0005
angle=0 # A link says that I have to comment these all
saturation = 1.5 # This on 
exposure = 1.5  # and this one too
hue=.1 # Should I change it to 0 too?

但是有这个链接说我必须发表评论hue,saturation,angle,exposure等。我想知道：

  1. 我是否必须将图像保存为目录中的灰度或代码会自行完成？
  2. 除了设置之外，还必须更改其他一些配置channels=1？此链接hue也建议设置为 0
  3. 我是否需要修改一些处理加载此链接中给出的图像作为函数的load_data_detection函数
4

2 回答 2

0

在线和官方存储库中有帖子建议编辑频道，色调值等。

这是适用于 SURE（我尝试过）的方法，对于训练集，要么拍摄灰度图像，要么使用 openCV 将 RGB 转换为灰度。

cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

按照您对 RGB 的要求训练 YOLOv4，即不要弄乱通道、色调等。

注意：不要忘记像通常对 RGB 图像所做的那样设置步骤、批量大小、通道数等，只需在灰度图像而不是 RGB 上训练模型。

很少有个人观察，我找不到的理论解释：

  1. YOLOv4 在 RGB 图像上训练，不适用于黑白/灰度图像。
  2. YOLOv4 在黑白/灰度图像上训练，不适用于 RGB。

希望这可以帮助。

编辑：我尚未验证这是否会比使用减少通道训练的模型在计算上更昂贵。尽管它表明不会减少/改善推理时间。

于 2021-04-17T17:52:21.177 回答
0

只需更改channels=1配置文件即可。如果不是，则注释掉其他参数，如angle,hue,exposure,saturation重试

于 2021-04-04T04:53:02.410 回答