Skip to content

microsoft/X-Decoder

 
 

Repository files navigation

X-Decoder Suite for Segmentation In the Wild

Organizers:

🔥 News

  • [2023.03.01] The Segmentation in the Wild Challenge had been launched and ready for submitting results!
  • [2023.02.28] We release Segmentation In the Wild dataset and evaluation code.

🎶 Introduction

seginw_allfig

The branch includes:

  • Dataset The download link for SGinW dataset;
  • Evaluation On the fly evaluation code supported X-Decoder evaluation.

Getting Started

Installation

pip3 install torch==1.13.1 torchvision==0.14.1 --extra-index-url https://download.pytorch.org/whl/cu113
python -m pip install 'git+https://github.com/MaureenZOU/detectron2-xyz.git'
pip install git+https://github.com/cocodataset/panopticapi.git
python -m pip install -r requirements.txt

# sometime this script will report an error when download the zip, please simply copy the command (This is cuased by bad character embedding)
sh install_cococapeval.sh

Download

Please download the dataset HERE!

.
└── seginw/
    ├── Airplane-Parts/
    │   ├── train/
    │   │   ├── *.jpg
    │   │   └── _annotations_min1cat.coco.json
    │   ├── train_10shot/
    │   │   └── ...
    │   └── valid/
    │       └── ...
    ├── Bottles/
    │   └── ...
    └── ...

Evaluation

  • Evaluate under Framework X-Decoder
mpirun -n 8 python eval.py evaluate --conf_files configs/xdecoder/svlp_focalt_lang.yaml  --overrides WEIGHT /pth/to/ckpt

Note: Due to zero-padding, filling a single gpu with multiple images may decrease the performance.

  • Evaluate Using Json File

Model Zoo

ADE ADE-full SUN SCAN SCAN40 Cityscape BDD
model ckpt PQ AP mIoU mIoU mIoU PQ mIoU mIoU PQ mAP mIoU PQ mIoU
X-Decoder BestSeg Tiny 19.1 10.1 25.1 6.2 35.7 30.3 38.4 22.4 37.7 18.5 50.2 16.9 47.6

Dataset Statistics

Screenshot 2023-02-28 at 8 46 56 AM

Submission Format

Please refer to detailed format HERE : )

.
├── ade.json/
│   └── {"ADE150-mIoU": "x", "ADE150-PQ": "x", "ADE150-mAP": "x", "ADE847-mIoU": "x"} 
├── seginw_Airplane-Parts_val.json/
│   └── coco format
├── seginw_Bottles_val.json
├── seginw_Brain-Tumor_val.json
├── seginw_Chicken_val.json
└── ...

Citation

@article{zou2022xdecoder,
  author      = {Zou, Xueyan and Dou, Zi-Yi and Yang, Jianwei and Gan, Zhe and Li, Linjie and Li, Chunyuan and Dai, Xiyang and Wang, Jianfeng and Yuan, Lu and Peng, Nanyun and Wang, Lijuan and Lee, Yong Jae and Gao, Jianfeng},
  title       = {Generalized Decoding for Pixel, Image and Language},
  publisher   = {arXiv},
  year        = {2022},
}

About

[CVPR 2023] Official Implementation of X-Decoder for generalized decoding for pixel, image and language

Resources

License

Code of conduct

Security policy

Stars

Watchers

Forks

Packages

No packages published

Languages