MP-Former

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

News

The installation is exactly same as Mask2Former

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

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

bash eval.sh

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.

Name		Name	Last commit message	Last commit date
Latest commit History 10 Commits
configs		configs
datasets		datasets
demo		demo
demo_video		demo_video
mask2former		mask2former
mask2former_video		mask2former_video
tools		tools
ADVANCED_USAGE.md		ADVANCED_USAGE.md
CODE_OF_CONDUCT.md		CODE_OF_CONDUCT.md
CONTRIBUTING.md		CONTRIBUTING.md
GETTING_STARTED.md		GETTING_STARTED.md
INSTALL.md		INSTALL.md
LICENSE		LICENSE
MODEL_ZOO.md		MODEL_ZOO.md
README.md		README.md
cog.yaml		cog.yaml
eval.sh		eval.sh
eval_batch.py		eval_batch.py
predict.py		predict.py
requirements.txt		requirements.txt
run_50ep_no_noise_all_ly.sh		run_50ep_no_noise_all_ly.sh
run_50ep_no_noise_all_ly_panoptic.sh		run_50ep_no_noise_all_ly_panoptic.sh
train_net.py		train_net.py
train_net_video.py		train_net_video.py