unet

UNet: Optimized for AMD ROCm

UNet-S5-D16 is a convolutional encoder-decoder network for semantic segmentation — it classifies every pixel in an image into one of 19 urban-scene categories (road, sidewalk, building, person, car, etc.). This repository packages inference for semantic segmentation using ONNX Runtime, exported and validated for AMD ROCm so it runs efficiently on AMD GPUs, CPUs, and NPUs.

This is based on the implementation of UNet found here. This repository contains configurations and scripts optimized for AMD® ROCm™ platforms. You can use the unet AMD scripts to reproduce results or export with custom configurations. More details on model performance can be found here.

---

Task Overview

Task: Cityscapes semantic segmentation (19 classes)

Dataset: Cityscapes val split (500 images), downloaded automatically from HuggingFace; benchmark inputs from UrbanSyn

Output metrics: mIoU (mean Intersection-over-Union across 19 classes), per-class IoU, pixel accuracy

---

AMD ROCm Optimization

This model export has been adapted and validated for AMD Instinct™ / Radeon™ GPUs running ROCm, as well as AMD CPUs and AMD Ryzen AI NPUs. Key points:

• Validated backends: ONNX Runtime across CPU (FP32), GPU (MIGraphX — FP32/FP16/BF16/INT8), and NPU (VitisAI — FP32/FP16/BF16/INT8).
• No code changes required versus the upstream UNet-S5-D16 (mmsegmentation) implementation — only environment/runtime configuration differs.
• All GPU precisions use a single FP32 ONNX model with runtime quantization via MIGraphX EP options.

---

Getting Started

For setup instructions, evaluation scripts, and custom configuration options, see the unet on GitHub.

---

Model Details

Model Type: Semantic segmentation (convolutional encoder-decoder)

Base Model: UNet-S5-D16 (mmsegmentation)

Model Stats:

• Input (input): (1, 3, 512, 1024) float32
• Output (output): (1, 19, 512, 1024) float32
• Precision tested: FP32 (CPU); FP32, FP16, BF16, INT8 (GPU)

---

Accuracy Pipeline

Higher mIoU means predicted pixel labels agree more closely with ground truth across all 19 Cityscapes classes — 100% is perfect overlap, 0% is no agreement. Published paper mIoU for UNet-S5-D16 on Cityscapes is 69.10%.

Metrics Explained

Accuracy Results

Cityscapes val mIoU — UNet-S5-D16 (paper: 69.10%):

<!-- accuracy-table-start --> <!-- accuracy-table-end -->

---

Dig Deeper

Want to explore the full evaluation scripts, config options, and other AMD-optimized model examples?

📂 View the full project on GitHub

The GitHub repository includes:

• Setup and prerequisites for ROCm environments
• Cityscapes val and UrbanSyn dataset staging scripts
• Full mIoU evaluation pipeline via mmsegmentation Runner
• Benchmarking and profiling scripts across CPU, GPU, and NPU