semantic-segmentation-adas-0001

February 10, 2022 · View on GitHub

Use Case and High-Level Description

This is a segmentation network to classify each pixel into 20 classes:

  • road
  • sidewalk
  • building
  • wall
  • fence
  • pole
  • traffic light
  • traffic sign
  • vegetation
  • terrain
  • sky
  • person
  • rider
  • car
  • truck
  • bus
  • train
  • motorcycle
  • bicycle
  • ego-vehicle

Example

Specification

MetricValue
Image size2048x1024
GFlops58.572
MParams6.686
Source frameworkCaffe*

Accuracy

The quality metrics calculated on 2000 images:

LabelIOU
mean0.6907
Road0.910379
Sidewalk0.630676
Building0.860139
Wall0.424166
Fence0.592632
Pole0.559078
Traffic Light0.654779
Traffic Sign0.648217
Vegetation0.882593
Terrain0.620521
Sky0.976889
Person0.711653
Rider0.612787
Car0.877892
Truck0.674829
Bus0.743752
Train0.358641
Motorcycle0.600701
Bicycle0.622246
Ego-Vehicle0.852932
  • IOU=TP/(TP+FN+FP), where:
    • TP - number of true positive pixels for given class
    • FN - number of false negative pixels for given class
    • FP - number of false positive pixels for given class

Inputs

The blob with BGR image and the shape 1, 3, 1024, 2048 in the format B, C, H, W, where:

  • B – batch size
  • C – number of channels
  • H – image height
  • W – image width

Outputs

The net output is a blob with the shape 1, 1, 1024, 2048 in the format B, C, H, W. It can be treated as a one-channel feature map, where each pixel is a label of one of the classes.

Demo usage

The model can be used in the following demos provided by the Open Model Zoo to show its capabilities:

[*] Other names and brands may be claimed as the property of others.