/Matting-Anything

Matting Anything Model (MAM), an efficient and versatile framework for estimating the alpha matte of any instance in an image with flexible and interactive visual or linguistic user prompt guidance.

Primary LanguagePythonMIT LicenseMIT

Matting Anything

YouTube HuggingFace Space Framework: PyTorch License

Jiachen Li, Jitesh Jain, Humphrey Shi

[Project page] [ArXiv] [Pdf] [Video] [Demo]

Updates

Contents

Matting Anything

Abstract

In this paper, we propose the Matting Anything Model (MAM), an efficient and versatile framework for estimating the alpha matte of any instance in an image with flexible and interactive visual or linguistic user prompt guidance. MAM offers several significant advantages over previous specialized image matting networks: (i) MAM is capable of dealing with various types of image matting, including semantic, instance, and referring image matting with only a single model; (ii) MAM leverages the feature maps from the Segment Anything Model (SAM) and adopts a lightweight Mask-to-Matte (M2M) module to predict the alpha matte through iterative refinement, which has only 2.7 million trainable parameters. (iii) By incorporating SAM, MAM simplifies the user intervention required for the interactive use of image matting from the trimap to the box, point, or text prompt. We evaluate the performance of MAM on various image matting benchmarks, and the experimental results demonstrate that MAM achieves comparable performance to the state-of-the-art specialized image matting models under different metrics on each benchmark. Overall, MAM shows superior generalization ability and can effectively handle various image matting tasks with fewer parameters, making it a practical solution for unified image matting.

Architecture


The MAM architecture consists of a pre-trained SAM and an M2M module. Given an input image I, SAM generates the mask prediction for the target instance based on the box or point user prompt. The M2M module takes the concatenated inputs, including the image, mask, and feature maps, and produces multi-scale predictions αos8, αos4, and αos1. The iterative refinement process, detailed in Section 3, progressively improves the precision of the final meticulous alpha matte α, incorporating information from the multi-scale outputs.

Visualization


We provide visualizations of the alpha matte predictions from SAM and MAM. Notably, we emphasize the differences in the red boxes. The visualizations demonstrate that MAM achieves improved predictions in the transition areas even without the trimap guidance. Additionally, MAM effectively addresses some of the holes present in the mask predictions generated by SAM. These visual comparisons highlight the superior performance of MAM in refining and enhancing the quality of alpha matte predictions.

Installation

Please refer to Installation Instructions for complete installation instructions for MAM.

Getting Started

Please refer to Getting Started for dataset preparation, training, and inference details of MAM.

Third-Party Projects

Citation

@article{li2023matting,
      title={Matting Anything},
      author={Jiachen Li and Jitesh Jain and Humphrey Shi},
      journal={arXiv: 2306.05399}, 
      year={2023}
    }

Acknowledgement

We thank the authors of SAM, Grounded-SAM, MGMatting, and InstMatt for releasing the codebases.