Home

Awesome

SOTR: Segmenting Objects with Transformers [ICCV 2021]

By Ruohao Guo, Dantong Niu, Liao Qu, Zhenbo Li

Introduction

This is the official implementation of SOTR.

<img src="images/overview.png" alt="image" style="zoom:60%;" />

Models

COCO Instance Segmentation Baselines with SOTR

Namemask APAP<sub>50</sub>AP<sub>75</sub>AP<sub>S</sub>AP<sub>M</sub>AP<sub>L</sub>download
SOTR_R5039.660.742.610.358.772.1model
SOTR_R10140.261.243.410.259.073.1model
SOTR_R101_DCN42.063.345.511.460.774.5model

Note: The area of AP<sub>S</sub>, AP<sub>M</sub> and AP<sub>L</sub> are calculated by segmentation mask without using bbox information.

Installation & Quick start

https://github.com/easton-cau/SOTR
cd SOTR
python setup.py build develop

Acknowledgement

Thanks Detectron2 and AdelaiDet contribution to the community!

The work is supported by National Key R&D Program of China (2020YFD0900204) and Key-Area Research and Development Program of Guangdong Province China (2020B0202010009).

FAQ

If you want to improve the usability or any piece of advice, please feel free to contant directly (ruohguo@foxmail.com).

Citation

Please consider citing our paper in your publications if the project helps your research. BibTeX reference is as follow.

@inproceedings{guo2021sotr,
  title={SOTR: Segmenting Objects with Transformers},
  author={Guo, Ruohao and Niu, Dantong and Qu, Liao and Li, Zhenbo},
  booktitle={Proceedings of the IEEE/CVF International Conference on Computer Vision},
  pages={7157--7166},
  year={2021}
}