Algoritmo leve de classificação de defeitos em painéis TFT-LCD baseado em Swin Transformer

XIA Yan ,  

LUO Chen ,  

ZHOU Yijun ,  

JIA Lei ,  

摘要

Na detecção de defeitos em painéis TFT-LCD, o fundo do objeto de detecção é complexo, os defeitos são sutis e variados, e a produção industrial exige alta capacidade em tempo real. Os algoritmos tradicionais de classificação de defeitos frequentemente não conseguem equilibrar precisão e velocidade, tornando-os inadequados para aplicação na produção real. Para equilibrar a precisão e a velocidade na classificação de defeitos de painéis TFT-LCD, propõe-se um modelo de classificação de imagens leve baseado no Swin Transformer. Primeiramente, realiza-se a fusão de tokens nos mapas de características de cada camada de entrada do modelo para reduzir a carga computacional, aumentando a leveza do modelo. Em seguida, é introduzido um módulo de convolução separável em profundidade para ajudar o modelo a aumentar o viés indutivo convolucional, mitigando a dependência do modelo em grandes volumes de dados. Por fim, utiliza-se o método de destilação do conhecimento para superar a redução de precisão decorrente da leveza do modelo. Experimentos no conjunto de dados próprio de classificação de defeitos em painéis TFT-LCD mostram que o modelo melhorado proposto reduz o cálculo de FLOPs em 2,6 bilhões em comparação ao modelo base, melhora a velocidade em 17% e perde apenas 1,3% de precisão Top-1, apresentando um equilíbrio mais equilibrado entre precisão e velocidade em relação a outros modelos principais de classificação de imagens em conjuntos de dados próprios e públicos.

关键词

TFT-LCD;Transformer;classificação de imagens;visão computacional

阅读全文

以上内容由讯飞翻译自动生成,翻译内容仅供参考。对于因使用本网站翻译内容产生的相关后果,本网站不承担任何商业和法律责任。

The above content is generated by Large Model Translation. The translated content is for reference only. We do not assume any commercial or legal responsibilty for any consequences arising from the use of our website