Skip to content
/ OASS Public

[ECCV2024] Occlusion-Aware Seamless Segmentation

Notifications You must be signed in to change notification settings

yihong-97/OASS

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

19 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Occlusion-Aware Seamless Segmentation

ECCV 2024

Yihong Cao1*, Jiaming Zhang2*, Hao Shi3, Kunyu Peng2, Yuhongxuan Zhang1, Hui Zhang1†, Rainer Stiefelhagen2, Kailun Yang1†
1Hunan University, 2Karlsruhe Institute of Technology, 3Zhejiang University

OASS [PDF]

Datasets

Considering the scalability of future work, we annotated 100 images containing 2,960 objects across various classes in the evaluation set, following the format of Cityscapes. For more details, please visit the BlendPASS homepage or download directly here.

KITTI360-APS-to-BlendPASS

Due to inconsistent class structures between KITTI360-APS and BlendPASS, we coarsely aligned the 8 class targets from BlendPASS to match the 7 classes in KITTI360-APS. The converted dataset can be downloaded directly here (code: oass).

Usage

Installation

pytorch==1.7.1
torchvision==0.8.2
torchaudio==0.7.2
cudatoolkit=10.2
mmcv-full==1.6.2

To facilitate your testing, we have packaged our dependencies. You can download them here and extract directly into your Anaconda virtual environment folder.

Datasets

Source: KITTI360-APS

Please refer to the homepage of KITTI360-APS. We also provide the dataset, which is available here.

Target: BlendPASS

Train

python run_experiments.py --config configs/unmaskformer/Training_OASS_UnmaskFormer.py

Evaluation

Our pretrained model is available at here.

python run_experiments.py --config configs/unmaskformer/Testing_UnmaskFormer.py

🤝 Publication:

Please consider referencing this paper if you use the code or data from our work. Thanks a lot :)

@inproceedings{cao2024oass,
  title={Occlusion-Aware Seamless Segmentation},
  author={Yihong Cao and Jiaming Zhang and Hao Shi and Kunyu Peng and Yuhongxuan Zhang and Hui Zhang and Rainer Stiefelhagen and Kailun Yang},
  booktitle={European Conference on Computer Vision (ECCV)},
  year={2024}
}

Acknowledgement:

This project is based on the following open-source projects. We thank their authors for making the source code publically available.

License

This project is released under the Apache License 2.0, while some specific features in this repository are with other licenses. Please refer to LICENSES.md for the careful check, if you are using our code for commercial matters.

About

[ECCV2024] Occlusion-Aware Seamless Segmentation

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages