计算机视觉错误,应该如何避免它们

人是不完美的,我们经常在程序中犯错误。有时这些错误很容易发现:你的代码根本不能工作,你的应用程序崩溃等等。但是有些bug是隐藏的,这使得它们更加危险。

在解决深度学习问题时,由于一些不确定性,很容易出现这种类型的bug:很容易看到web应用端点路由请求是否正确,而不容易检查你的梯度下降步骤是否正确。然而,在DL从业者生涯中有很多错误是可以避免的。

8个深度学习/计算机视觉错误,应该如何避免它们

我想分享一些我的经验,关于我在过去两年的计算机视觉工作中看到或制造的错误。我在会议上谈到过这个话题,很多人在会后告诉我:“是的,伙计,我也有很多这样的错误。”我希望我的文章可以帮助你至少避免其中的一些问题。

1.翻转图像和关键点

假设一个关键点检测问题的工作。它们的数据看起来像图像和一系列关键点元组,例如[(0,1),(2,2)],其中每个关键点是一对x和y坐标。

让我们对这个数据实现一个基本的数据增强:

看起来好像是正确的,嗯,让我们把结果可视化一下:

8个深度学习/计算机视觉错误,应该如何避免它们

不对称看起来很奇怪!如果我们检查极值的情况呢?

out:

程序报错了!这是一个典型的差一误差。正确的代码是这样的:

我们可以通过可视化来检测这个问题,而在x = 0点的单元测试也会有帮助。

2.还是关键点问题

即使在上述错误被修复之后,仍然存在问题。现在更多的是语义上的问题,而不仅仅是代码上的问题。

假设需要增强具有两只手掌的图像。看起来好像没问题-左右翻转后手还是手。

8个深度学习/计算机视觉错误,应该如何避免它们

但是等等!我们对我们拥有的关键点语义一无所知。如果这个关键点的意思是这样的:

8个深度学习/计算机视觉错误,应该如何避免它们

这意味着增强实际上改变了语义:左变成右,右变成左,但我们不交换数组中的关键点索引。它会给训练带来大量的噪音和更糟糕的度量。

我们应该吸取教训:

  • 在应用增强或其他特性之前,要了解和考虑数据结构和语义;
  • 保持你的实验原子性:添加一个小的变化(例如一个新的变换),如果分数已经提高,检查它如何进行和合并。

3.编码自定义损失函数

熟悉语义分割问题的人可能知道IoU度量。不幸的是,我们不能直接用SGD来优化它,所以常用的方法是用可微损失函数来近似它。让我们编码实现一个!

看起来不错,让我们测试一下:

在x1中,我们计算了与正确数据完全不同的数据的损失,而x2则是非常接近正确数据的数据损失结果。我们期望x1很大因为预测很糟糕,x2应该接近0。但是结果与我期望的有差别,哪里出现错误了呢?

上面的函数是度量的一个很好的近似。度量不是一种损失:它通常(包括这种情况)越高越好。当我们使用SGD最小化损失时,我们应该做一些改变:

这些问题可以从两个方面来确定:

  • 编写一个单元测试来检查损失的方向
  • 运行健全性检查

4.当我们遇到Pytorch的时候

假设有一个预先训练好的模型。编写基于ceevee API的Predictor 类。

这个代码正确吗?也许!对于某些模型来说确实是正确的。例如,当模型没有dropout或norm 层,如torch.nn.BatchNorm2d。

但是对于大多数计算机视觉应用来说,代码忽略了一些重要的东西:转换到评估模式。

如果试图将动态PyTorch图转换为静态PyTorch图,这个问题很容易意识到。torch.jit模块用于这种转换。

一个简单的解决办法:

torch.jit.trace运行模型几次并比较结果。 然而torch.jit.trace并不是万能的,你应该了解并记住。

5.复制粘贴问题

很多东西都是成对存在的:训练和验证、宽度和高度、纬度和经度……如果你仔细阅读,你会很容易发现一个bug是由某一个成员中复制粘贴到另外一个成员中引起的:

不仅仅是我犯了愚蠢的错误,例如。流行的albumentations库中也有类似的问题。

不过别担心,现在已经修复好了。

如何避免?尽量以不需要复制和粘贴的方式编写代码。

下面这种编程方式不是一个好的方式:

而下面的方式看起来好多了:

6.正确的数据类型

让我们编写一个新的增强:

8个深度学习/计算机视觉错误,应该如何避免它们

图像已被更改。这是我们所期望的吗?嗯,可能修改得有点过了。 这里有一个危险的操作:将float32转换为uint8。它可能会导致溢出:

看起来好多了,是吧?

顺便说一句,还有一种方法可以避免这个问题:不要重造轮子,不要从头开始编写增强代码,而是使用现有的增强,比如:albumentations.augmentations.transforms.GaussNoise。

我曾经犯过另一个同样的错误。

这里出了什么问题?首先,用三次样条插值调整mask的大小是一个坏主意。与转换float32到uint8的问题是一样的:三次样条插值的输出值会大于输入值,会导致溢出。

8个深度学习/计算机视觉错误,应该如何避免它们

我在做可视化的时候发现了这个问题。在你的训练循环中到处使用断言也是一个好主意。

7. 拼写错误发生

假设需要对全卷积网络(如语义分割问题)和一个巨大的图像进行推理。该图像是如此巨大,没有机会把它放在你的GPU上 -例如,它可以是一个医疗或卫星图像。

在这种情况下,可以将图像分割成网格,独立地对每一块进行推理,最后合并。此外,一些预测交叉可能有助于平滑边缘的伪影

让我们编码实现吧!

有一个符号输入错误,可以很容易地找到它,检查代码是否正确:

__call__方法的正确版本如下:

如果你仍然不知道问题是什么,注意行weights[a:b, c:d, :] += 1。

8.Imagenet归一化

当一个人需要做迁移学习时,用训练Imagenet时的方法将图像归一化通常是一个好主意。

让我们使用熟悉的albumentations来实现:

现在是时候训练一个网络并对单个图像进行拟合——正如我所提到的,这是一种很好的调试技术:

8个深度学习/计算机视觉错误,应该如何避免它们

曲率看起来很好,但是-300不是我们期望的交叉熵的损失值。是什么问题?

归一化处理图像效果很好,但掩码需要缩放到[0,1]之间。

在训练循环时一个简单运行断言(例如assert mask.max() <= 1)可以很快地检测到问题。同样,也可以是单元测试。


分享到:


相關文章: