边缘检测-Tiny and Efficient Model for the Edge Detection Generalization

源代码: https://github.com/xavysp/TEED

论文地址：https://arxiv.org/pdf/2308.06468.pdf

大多数高级计算机视觉任务依赖于低级图像操作作为其初始过程。边缘检测、图像增强和超分辨率等操作为更高级的图像分析提供了基础。在这项工作中，我们考虑三个主要目标来解决边缘检测:简单，效率和泛化，因为当前最先进的(SOTA)边缘检测模型的复杂性增加了，以获得更好的准确性。

为了实现这一点，我们提出了微小和高效的边缘检测器(TEED)是一种轻量级的卷积神经网络，只有58K个参数，不到最先进模型的0.2%。在BIPED数据集上的训练用时不到30分钟，每个epoch用时不到5分钟。我们提出的模型易于训练，并且在最初的几个epoch内快速收敛，同时预测的边缘图清晰且质量高。此外，我们提出了一个新的数据集来测试边缘检测的泛化，它包括用于边缘检测和图像分割的常用图像的样本。