← Open Source
uber-research

UPSNet

UPSNet: A Unified Panoptic Segmentation Network.

Model DevelopmentArchitecturePython
Open on GitHub
Momentum
+0stars in 24 hours0.0%
646
Stars
112
Forks
+0
This week
0
Contributors
Created 2019-03-02 · Updated 2026-10-01 · #15575 today

uber-research/UPSNet is ranked 15,575 with 646 stars.

Same-category comparison

uber-research/UPSNet is ranked 15,575 among these 8 peers; AnyLoc/AnyLoc is ranked 15,577.

Same-category comparison
NameRankMetrics
webis-de/small-text15,565
Language:
Python
Stars:
647
Forks:
78
Description:
Active Learning for Text Classification in Python.
BlackHC/tfpyth15,567
Language:
Python
Stars:
646
Forks:
95
Description:
Putting TensorFlow back in PyTorch, back in TensorFlow (differentiable TensorFlow PyTorch adapters).
SJTUjinmao/ArkEval15,568
Language:
Python
Stars:
646
Forks:
16
Description:
ArkEval: Benchmarking and Evaluating Automated Code Repair for ArkTS.
maciejkula/rustlearn15,572
Language:
Rust
Stars:
646
Forks:
55
Description:
Machine learning crate for Rust.
AnyLoc/AnyLoc15,577
Language:
Python
Stars:
645
Forks:
66
Description:
AnyLoc: Universal Visual Place Recognition (RA-L 2023).
erdogant/bnlearn15,583
Language:
Jupyter Notebook
Stars:
645
Forks:
61
Description:
Python package for Causal Discovery by learning the graphical structure of Bayesian networks. Structure Learning, Parameter Learning, Inferences, Sampling methods.
huanghoujing/AlignedReID-Re-Production-Pytorch15,584
Language:
Python
Stars:
645
Forks:
189
Description:
Reproduce AlignedReID: Surpassing Human-Level Performance in Person Re-Identification, using Pytorch.
jvanvugt/pytorch-domain-adaptation15,586
Language:
Python
Stars:
645
Forks:
107
Description:
A collection of implementations of adversarial domain adaptation algorithms.
Key parameters and specifications
Key parameters and specifications
ParameterValue
LanguagePython
CategoryModel Development
SubcategoryArchitecture
KeywordsComputer vision
Stars646
Forks112
Contributors0
Created date2019-03-02
Updated date2026-10-01
Stars in 1 day0
Stars in 7 days0
Rank15,575
Previous rank15,384
Rank change-191
7-day stars rank15,067
Previous 7-day stars rank15,069
7-day stars rank change2
Total stars rank15,853
Previous total stars rank15,825
Total stars rank change-28
CountryUSA
README summary

UPSNet: A Unified Panoptic Segmentation Network Introduction UPSNet is initially described in a CVPR 2019 oral paper. Disclaimer This repository is tested under Python 3. And model training is done with 16 GPUs by using horovod. Licensed under the Uber Non-Commercial License. 8 | Requirements: Software We recommend using Anaconda3 as it already includes many common packages. Requirements: Hardware We recommend using 4~16 GPUs with at least 11 GB memory to train our model. Installation Clone this repo to

Run to build essential C++/CUDA modules and download pretrained model. For Cityscapes: Assuming you already downloaded Cityscapes dataset at and TrainIds label images are generated, please create a soft link by under , and run to prepare Cityscapes dataset for UPSNet. For COCO: Assuming you already downloaded COCO dataset at and have and folders under it, please create a soft link by under , and run to prepare COCO dataset for UPSNet. Training: Test: We provide serveral config files (16/4

GPUs for Cityscapes/COCO dataset) under upsnet/experiments folder. Model Weights The model weights that can reproduce numbers in our paper are available now. Please follow these steps to use them: Run to get trained model weights for Cityscapes and COCO.

Frequently asked questions

How many stars and forks are recorded?

uber-research/UPSNet has 646 stars and 112 forks.

What is the repository rank?

uber-research/UPSNet is ranked 15,575.

What language and category are recorded?

uber-research/UPSNet has 646 stars, uses Python and belongs to Model Development.

Which repository is closest in rank?

uber-research/UPSNet is compared with AnyLoc/AnyLoc, ranked 15,577 with 645 stars.

README

UPSNet: A Unified Panoptic Segmentation Network

Introduction

UPSNet is initially described in a CVPR 2019 oral paper.

Disclaimer

This repository is tested under Python 3.6, PyTorch 0.4.1. And model training is done with 16 GPUs by using horovod. It should also work under Python 2.7 / PyTorch 1.0 and with 4 GPUs.

License

© Uber, 2018-2019. Licensed under the Uber Non-Commercial License.

Citing UPSNet

If you find UPSNet is useful in your research, please consider citing:

@inproceedings{xiong19upsnet,
    Author = {Yuwen Xiong, Renjie Liao, Hengshuang Zhao, Rui Hu, Min Bai, Ersin Yumer, Raquel Urtasun},
    Title = {UPSNet: A Unified Panoptic Segmentation Network},
    Conference = {CVPR},
    Year = {2019}
}

Main Results

COCO 2017 (trained on train-2017 set)

test split PQ SQ RQ PQTh PQSt
UPSNet-50 val 42.5 78.0 52.4 48.5 33.4
UPSNet-101-DCN test-dev 46.6 80.5 56.9 53.2 36.7

Cityscapes

PQ SQ RQ PQTh PQSt
UPSNet-50 59.3 79.7 73.0 54.6 62.7
UPSNet-101-COCO (ms test) 61.8 81.3 74.8 57.6 64.8

Requirements: Software

We recommend using Anaconda3 as it already includes many common packages.

Requirements: Hardware

We recommend using 4~16 GPUs with at least 11 GB memory to train our model.

Installation

Clone this repo to $UPSNet_ROOT

Run init.sh to build essential C++/CUDA modules and download pretrained model.

For Cityscapes:

Assuming you already downloaded Cityscapes dataset at $CITYSCAPES_ROOT and TrainIds label images are generated, please create a soft link by ln -s $CITYSCAPES_ROOT data/cityscapes under UPSNet_ROOT, and run init_cityscapes.sh to prepare Cityscapes dataset for UPSNet.

For COCO:

Assuming you already downloaded COCO dataset at $COCO_ROOT and have annotations and images folders under it, please create a soft link by ln -s $COCO_ROOT data/coco under UPSNet_ROOT, and run init_coco.sh to prepare COCO dataset for UPSNet.

Training:

python upsnet/upsnet_end2end_train.py --cfg upsnet/experiments/$EXP.yaml

Test:

python upsnet/upsnet_end2end_test.py --cfg upsnet/experiments/$EXP.yaml

We provide serveral config files (16/4 GPUs for Cityscapes/COCO dataset) under upsnet/experiments folder.

Model Weights

The model weights that can reproduce numbers in our paper are available now. Please follow these steps to use them:

Run download_weights.sh to get trained model weights for Cityscapes and COCO.

For Cityscapes:

python upsnet/upsnet_end2end_test.py --cfg upsnet/experiments/upsnet_resnet50_cityscapes_16gpu.yaml --weight_path ./model/upsnet_resnet_50_cityscapes_12000.pth
python upsnet/upsnet_end2end_test.py --cfg upsnet/experiments/upsnet_resnet101_cityscapes_w_coco_16gpu.yaml --weight_path ./model/upsnet_resnet_101_cityscapes_w_coco_3000.pth

For COCO:

python upsnet/upsnet_end2end_test.py --cfg upsnet/experiments/upsnet_resnet50_coco_16gpu.yaml --weight_path model/upsnet_resnet_50_coco_90000.pth
python upsnet/upsnet_end2end_test.py --cfg upsnet/experiments/upsnet_resnet101_dcn_coco_3x_16gpu.yaml --weight_path model/upsnet_resnet_101_dcn_coco_270000.pth
Nearby in the ranking
Similar items