Skip to content

[CVPR 2023] Official implementation of the paper: MP-Former: Mask-Piloted Transformer for Image Segmentation

License

Notifications You must be signed in to change notification settings

IDEA-Research/MP-Former

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

MP-Former

This is the official implementation of the paper "MP-Former: Mask-Piloted Transformer for Image Segmentation". Accepted to CVPR 2023.

News

  • 8/19/2023: Code is released.

Install

The installation is exactly same as Mask2Former

Train

The following command is for no noise and all-layer MP training setting for instance segmentation.

bash run_50ep_no_noise_all_ly.sh

The following command is for no noise and all-layer MP training setting for panoptic segmentation.

bash run_50ep_no_noise_all_ly_panoptic.sh

Eval

You can set "weights" to the path of the checkpoint.

bash eval.sh

Model

This checkpoint is the 12-epoch checkpoint (last row in Table 7 of the paper). You are expected to have AP: 40.15 if you evaluate it.

About

[CVPR 2023] Official implementation of the paper: MP-Former: Mask-Piloted Transformer for Image Segmentation

Resources

License

Code of conduct

Stars

Watchers

Forks

Packages

No packages published

Languages