← all repositories
bubbliiiing/deeplabv3-plus-pytorch

DeepLabv3+ for developers who just want to train already

This repo wraps DeepLabv3+ into a practical PyTorch training kit with pre-trained weights and mIOU evaluation, letting you fine-tune on custom data without reimplementing the paper.

1.3k stars Python Computer VisionML Frameworks
deeplabv3-plus-pytorch
Not currently ranked — collecting fresh signals.
star history

What it does

It is a PyTorch reimplementation of the DeepLabv3+ semantic-segmentation architecture, offering both MobileNetV2 and Xception backbones. The repo includes training, prediction, and mIOU evaluation scripts, plus pre-trained weights on the VOC12+SBD dataset. You bring your own labeled data in VOC format and swap the backbone to trade accuracy for speed.

The interesting bit

The value is not novelty; it is completeness. The author treats the repo as part of an assembly line—sibling repos cover U-Net, PSPNet, and HRNet—so the API and data flow stay predictable if you jump between models. It also supports multi-GPU training and adaptive learning-rate scaling based on batch size, practical touches often missing from bare-bones research ports.

Key highlights

  • Ships with pre-trained weights for MobileNetV2 (72.59 mIOU) and Xception (76.95 mIOU) on VOC-Val12 at 512×512.
  • Supports both backbones with configurable downsample factors (8 or 16).
  • Includes utilities for folder-based batch inference, video segmentation, and FPS testing.
  • Multi-GPU training, step/cosine LR schedules, and adaptive LR by batch size.
  • Heavily commented code and linked BiliBili video tutorials (Chinese language).

Caveats

  • The environment pins torch==1.2.0, a notably old PyTorch release.
  • Weights and datasets are distributed exclusively via Baidu Netdisk with extraction codes, which may be inaccessible or cumbersome depending on your region.

Verdict

Grab this if you need a plug-and-play DeepLabv3+ training pipeline for custom VOC-format segmentation data and you value working code over bleeding-edge framework versions. Skip it if you are looking for architectural innovations or a modern, pip-installable package.

Frequently asked

What is bubbliiiing/deeplabv3-plus-pytorch?
This repo wraps DeepLabv3+ into a practical PyTorch training kit with pre-trained weights and mIOU evaluation, letting you fine-tune on custom data without reimplementing the paper.
Is deeplabv3-plus-pytorch open source?
Yes — bubbliiiing/deeplabv3-plus-pytorch is open source, released under the MIT license.
What language is deeplabv3-plus-pytorch written in?
bubbliiiing/deeplabv3-plus-pytorch is primarily written in Python.
How popular is deeplabv3-plus-pytorch?
bubbliiiing/deeplabv3-plus-pytorch has 1.3k stars on GitHub.
Where can I find deeplabv3-plus-pytorch?
bubbliiiing/deeplabv3-plus-pytorch is on GitHub at https://github.com/bubbliiiing/deeplabv3-plus-pytorch.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.