Puzzle-CAM: Improved Localization Via Matching Partial And Full Features

By Sanghyun Jo, In-Jae Yu. 2021

131 citations recorded by the source corpus.

Weakly-supervised semantic segmentation (WSSS) is introduced to narrow the gap for semantic segmentation performance from pixel-level supervision to image-level supervision. Most advanced approaches are based on class activation maps (CAMs) to generate pseudo-labels to train the segmentation network. The main limitation of WSSS is that the process of generating pseudo-labels from CAMs that use an image classifier is mainly focused on the most discriminative parts of the objects. To address this issue, we propose Puzzle-CAM, a process that minimizes differences between the features from separate patches and the whole image. Our method consists of a puzzle module and two regularization terms to discover the most integrated region in an object. Puzzle-CAM can activate the overall region of an object using image-level supervision without requiring extra parameters. In experiments, Puzzle-CAM outperformed previous state-of-the-art methods using the same labels for supervision on the PASCAL VOC 2012 dataset. Code associated with our experiments is available at https://github.com/OFRIN/PuzzleCAM.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Yin Cui
  2. Tsung-Yi Lin
  3. Golnaz Ghiasi
  4. Xiuye Gu
  5. X. Wang
  6. Jan Kautz
  7. Jiarui Xu
  8. Sifei Liu

In Computer Vision and Pattern Recognition

  1. Ming-Hsuan Yang
  2. Luc Van Gool
  3. Dacheng Tao
  4. Trevor Darrell
  5. Chen Change Loy
  6. Alan L. Yuille
  7. Xiaogang Wang
  8. Jitendra Malik

Citations in the corpus