#290 · Primary category: Computer Vision

Efficient-Segmentation-Networks

camvid cityscapes computer-vision driving-scene-understanding efficient-segmentation-networks image-segmentation lightweight-semantic-segmentation neural-networks pytorch real-time-semantic-segmentation scene-understanding segmentation semantic-segmentation semantic-segmentation-models

Lightweight models for real-time semantic segmentationon PyTorch (include SQNet, LinkNet, SegNet, UNet, ENet, ERFNet, EDANet, ESPNet, ESPNetv2, LEDNet, ESNet, FSSNet, CGNet, DABNet, Fast-SCNN, ContextNet, FPENet, etc.)

Project last updated:07/25/24

GitHub Stars

1.0K

Forks

167

Contributors

1

License

MIT

Why we included this project

Choosing a lightweight segmentation network usually means juggling a dozen separate repos, each with its own quirks. This one collects many of the efficient encoder-decoder models people actually use for real-time work, from ENet and ERFNet to ESPNet, LEDNet, CGNet and Fast-SCNN, all written in a consistent PyTorch style. That consistency matters because you can compare backbones and swap them without relearning a new codebase each time. The code is set up around Cityscapes and CamVid, the standard driving-scene benchmarks, and bundles the loss functions, optimizers, normalization tricks, and learning-rate schedules you need to reproduce or adapt training runs. It is a reference toolbox rather than a polished product, so expect to read the code and fit it to your own pipelines, which suits researchers, students, and engineers prototyping segmentation on embedded or low-latency systems.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category