• Stars
    star
    401
  • Rank 106,944 (Top 3 %)
  • Language
    Python
  • License
    MIT License
  • Created over 1 year ago
  • Updated 6 months ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

[ICLR'24] Matcher: Segment Anything with One Shot Using All-Purpose Feature Matching

Matcher: Segment Anything with One Shot Using All-Purpose Feature Matching

Yang Liu1*,   Muzhi Zhu1*,   Hengtao Li1*,   Hao Chen1,   Xinlong Wang2,   Chunhua Shen1

1Zhejiang University,   2Beijing Academy of Artificial Intelligence

🚀 Overview

image

📖 Description

Powered by large-scale pre-training, vision foundation models exhibit significant potential in open-world image understanding. Even though individual models have limited capabilities, combining multiple such models properly can lead to positive synergies and unleash their full potential. In this work, we present Matcher, which segments anything with one shot by integrating an all-purpose feature extraction model and a class-agnostic segmentation model. Naively connecting the models results in unsatisfying performance, e.g., the models tend to generate matching outliers and false-positive mask fragments. To address these issues, we design a bidirectional matching strategy for accurate cross-image semantic dense matching and a robust prompt sampler for mask proposal generation. In addition, we propose a novel instance-level matching strategy for controllable mask merging. The proposed Matcher method delivers impressive generalization performance across various segmentation tasks, all without training. For example, it achieves 52.7% mIoU on COCO-20i for one-shot semantic segmentation, surpassing the state-of-the-art specialist model by 1.6%. In addition, our visualization results show open-world generality and flexibility on images in the wild.

🗓️ TODO

  • Online Demo
  • Release code and models

🖼️ Demo

One-Shot Semantic Segmantation

image

One-Shot Object Part Segmantation

image

Cross-Style Object and Object Part Segmentation

image

Controllable Mask Output

image

Video Object Segmentation

vos_demo.mp4

🎫 License

The content of this project itself is licensed under LICENSE.

🖊️ Citation

If you find this project useful in your research, please consider cite:

@article{liu2023matcher,
  title={Matcher: Segment Anything with One Shot Using All-Purpose Feature Matching},
  author={Liu, Yang and Zhu, Muzhi and Li, Hengtao and Chen, Hao and Wang, Xinlong and Shen, Chunhua},
  journal={arXiv preprint arXiv:2305.13310},
  year={2023}
}

More Repositories

1

AdelaiDet

AdelaiDet is an open source toolbox for multiple instance-level detection and recognition tasks.
Python
3,349
star
2

AdelaiDepth

This repo contains the projects: 'Virtual Normal', 'DiverseDepth', and '3D Scene Shape'. They aim to solve the monocular depth estimation, 3D scene reconstruction from single image problems.
Python
1,051
star
3

StyleDrop-PyTorch

This is an unofficial PyTorch implementation of StyleDrop: Text-to-Image Generation in Any Style.
Python
203
star
4

Poseur

[ECCV 2022] The official repo for the paper "Poseur: Direct Human Pose Regression with Transformers".
Python
176
star
5

AutoStory

143
star
6

DyCo3D

Python
119
star
7

GenPercept

GenPercept: Diffusion Models Trained with Large Data Are Transferable Visual Models
Python
112
star
8

FrozenRecon

[ICCV2023] 🧊FrozenRecon: Pose-free 3D Scene Reconstruction with Frozen Depth Models
Python
108
star
9

SegPrompt

Official Implementation of ICCV 2023 Paper - SegPrompt: Boosting Open-World Segmentation via Category-level Prompt Learning
Python
108
star
10

FreeCustom

[CVPR 2024] Official PyTorch implementation of FreeCustom: Tuning-Free Customized Image Generation for Multi-Concept Composition
Python
69
star
11

OIR

[ICLR 2024] Official PyTorch/Diffusers implementation of "Object-aware Inversion and Reassembly for Image Editing"
Python
69
star
12

RGM

68
star
13

GeoBench

A toolbox for benchmarking SOTA discriminative and generative geometry estimation models.
Python
42
star
14

GenDeF

38
star
15

DiverGen

DiverGen (CVPR 2024) & BSGAL (ICML 2024)
Python
32
star
16

partially-labelled

Learning to segment multi-organ and tumorsfrom multiple partially labeled datasets
19
star
17

FreeCompose

13
star
18

VFN

[ICLR 2024 Spotlight] The official repo for the paper "De novo Protein Design using Geometric Vector Field Networks".
Python
12
star
19

VLModel

Repo of HawkLlama.
Python
8
star
20

Depth3D

Python
7
star
21

FADiff

[ICML 2024] Floating Anchor Diffusion Model for Multi-motif Scaffolding
Python
6
star
22

STORY

4
star
23

LoRAPrune

Python
4
star
24

MovieDreamer

4
star
25

OIR-Diffusion

JavaScript
1
star