• Stars
    star
    434
  • Rank 100,274 (Top 2 %)
  • Language
    Python
  • License
    Other
  • Created over 1 year ago
  • Updated 3 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,375
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,063
star
3

MovieDreamer

249
star
4

StyleDrop-PyTorch

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

Poseur

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

AutoStory

144
star
7

GenPercept

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

DyCo3D

Python
121
star
9

FrozenRecon

[ICCV2023] ๐ŸงŠFrozenRecon: Pose-free 3D Scene Reconstruction with Frozen Depth Models
Python
117
star
10

SegPrompt

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

FreeCustom

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

OIR

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

RGM

69
star
14

GeoBench

A toolbox for benchmarking SOTA discriminative and generative geometry estimation models.
Python
49
star
15

GenDeF

38
star
16

DiverGen

DiverGen (CVPR 2024) & BSGAL (ICML 2024)
Python
35
star
17

LoRAPrune

Python
30
star
18

FreeCompose

Jupyter Notebook
23
star
19

SINE

[NeurIPS'24] A Simple Image Segmentation Framework via In-Context Examples
Python
21
star
20

VFN

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

partially-labelled

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

FADiff

[ICML 2024] Floating Anchor Diffusion Model for Multi-motif Scaffolding
Python
19
star
23

Depth3D

Python
10
star
24

VLModel

Repo of HawkLlama.
Python
10
star
25

STORY

4
star
26

OIR-Diffusion

JavaScript
1
star