Sewer-Transformer-ML

新型Vision Transformer模型提升下水道缺陷分类能力

研究人员开发了Sewer-Transformer-ML，一种新颖的Vision Transformer模型，用于下水道缺陷的多标签分类。该模型集成了多层次特征融合，并在Sewer-ML测试集上取得了最先进的性能，在$F2_{ ext{CIW}}$指标上显著优于第二名的方法。此外，还为资源受限环境引入了两种轻量级架构Sewer-MobileNet-ML和Sewer-Mobile-TransNet，在大幅减少参数量的同时保持了高精度。