DAFFnet: Clasificarea semințelor de soia bazată pe rețele de fuziune a caracteristicilor cu atenție dublă.
Clasificarea rapidă și precisă a semințelor de soia este necesară pentru controlul calității produsului.
Prezentăm un model deep-learning bazat pe imagini hiperspectrale, denumit Rețele de Fuziune a Caracteristicilor cu Atenție Duală (DAFFnet), care aplică în mod secvențial o Rețea Neuronală Convoluțională 3D (CNN) și o CNN 2D.
Un modul de mecanism de atenție prin fuziune în cadrul CNN 2D permite modelului să captureze atât informații despre caracteristici locale, cât și globale, combinându-se cu Modulul de Atenție pentru Blocurile de Convoluție (CBAM) și Transformer-ul Mobile Vision (MViT), depășind în performanță modelele convenționale de clasificare a imaginilor hiperspectrale în clasificarea semințelor..
Publică comentariul