Journal of Jianghan University (Natural Science Edition) ›› 2023, Vol. 51 ›› Issue (6): 63-71.doi: 10.16389/j.cnki.cn42-1737/n.2023.06.009

Previous Articles    

Fine-grained Image Classification Based on Mutual Channel Loss Data Augmentation Network

HU Xiaobin,PENG Taile   

  1. School of Computer Science and Technology,Huaibei Normal University,Huaibei 235000,Anhui,China
  • Published:2023-12-25
  • Contact: PENG Taile

Abstract: Finding discriminative local regions corresponding to fine-grained features is the key to solve fine-grained image classification problems. Recently, fine-grained classification by weakly supervised data augmentation network(WS-DAN)has achieved excellent results,but its single cross-entropy loss(CE-Loss)makes the network focus on global discriminative regions and misses some local discriminative regions. To address this problem,this paper proposed a data augmentation network (MC-DAN) based on mutual channel loss(MC-Loss),which could force feature channels belonging to the same class to be more discriminative. Second,a counterfactual attention mechanism(CAL)was introduced to encourage the network to learn more attention information by counterfactual intervention. In addition, an attention module combining spatial attention and channel attention was proposed to better focus on object regions in images. Comprehensive experiments on three public datasets showed that the method could effectively achieve classification.

Key words: fine-grained image classification, mutual channel loss, counterfactual attention learning, data augmentation

CLC Number: