Replacing convolution with morphological operations in trainable layers has received significant attention lately. Among the various strategies that have emerged, smooth morphological layers have shown strong potential and flexibility, as a single layer can behave either like a (pseudo-)erosion or a (pseudo-)dilation depending on the sign and value of its trainable control parameter. In this work, we build upon the so-called \(\mathcal {S}\) Morph layer by introducing a harmonized formulation that addresses previously identified asymptotic limitations when learning grayscale erosion and dilation. We also investigate and compare two strategies (a novel penalty term in the training loss and shared-weight layers) to improve the learning of grayscale opening and closing operations in two-layer networks. Finally, we evaluate the performance of this improved \(\mathcal {S}\) Morph layer on a salt-and-pepper denoising task in a four-layer network architecture, and compare it with other morphological and convolutional networks.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Improving Morphological Networks for Learning Image-to-Image Transforms

  • Antoine Bottenmuller,
  • Guillaume Tochon,
  • Romain Hermary,
  • Élodie Puybareau,
  • Gustavo Angulo

摘要

Replacing convolution with morphological operations in trainable layers has received significant attention lately. Among the various strategies that have emerged, smooth morphological layers have shown strong potential and flexibility, as a single layer can behave either like a (pseudo-)erosion or a (pseudo-)dilation depending on the sign and value of its trainable control parameter. In this work, we build upon the so-called \(\mathcal {S}\) Morph layer by introducing a harmonized formulation that addresses previously identified asymptotic limitations when learning grayscale erosion and dilation. We also investigate and compare two strategies (a novel penalty term in the training loss and shared-weight layers) to improve the learning of grayscale opening and closing operations in two-layer networks. Finally, we evaluate the performance of this improved \(\mathcal {S}\) Morph layer on a salt-and-pepper denoising task in a four-layer network architecture, and compare it with other morphological and convolutional networks.