使用 tensorflow object detection 进行 either or 检测

Question

我使用 Tensorflow 对象检测已经有一段时间了。我更像是一个用户，我真的不知道它是如何工作的。我想知道是否有可能训练它识别一个物体是某物而不是某物？例如，我想检测瓷砖上的裂缝。我可以使用对象检测来显示瓷砖的图像，它可以告诉我是否有裂缝（并显示位置），或者它会告诉我瓷砖上是否没有裂缝？

我尝试使用有缺陷和无缺陷的图片进行训练，使用 2 类（1 个表示有缺陷，1 个表示没有缺陷）。但是结果一直在一张图片中显示两者（如果图片有缺陷）。有没有办法只显示有缺陷的？

基本上我想做缺陷检查。这是 1 个缺陷的简单案例。但实际情况会有一些缺陷。 谢谢。

Answer 1

如果您只需要瓷砖的输入图像，无论是否有缺陷，您都不需要 class 表示没有缺陷。 API 为其他 class 之外的所有内容添加背景 class。 所以你只需要声明一个 class - 缺陷，并且没有被检测到的瓷砖没有缺陷。 所以在你的训练集中 - 简单地给出缺陷的边界框，没有缺陷的情况下没有边界框，然后你的模型应该学会检测缺陷，如上所述。