View source on GitHub
|
Semantic segmentation configuration definition.
Classes
class DataConfig: Input config for training.
class DenseFeatureConfig: Config for dense features, such as RGB pixels, masks, heatmaps.
class Evaluation: Evaluation config.
class ExportConfig: Model export config.
class Losses: Loss function config.
class MaskScoringHead: Mask Scoring head config.
class SegmentationHead: Segmentation head config.
class SemanticSegmentationModel: Semantic segmentation model config.
class SemanticSegmentationTask: The model config.
Functions
mnv2_deeplabv3_cityscapes(...): Image segmentation on cityscapes with mobilenetv2 deeplabv3.
mnv2_deeplabv3_pascal(...): Image segmentation on pascal with mobilenetv2 deeplabv3.
mnv2_deeplabv3plus_cityscapes(...): Image segmentation on cityscapes with mobilenetv2 deeplabv3plus.
seg_deeplabv3_pascal(...): Image segmentation on pascal voc with resnet deeplabv3.
seg_deeplabv3plus_cityscapes(...): Image segmentation on cityscapes with resnet deeplabv3+.
seg_deeplabv3plus_pascal(...): Image segmentation on pascal voc with resnet deeplabv3+.
seg_resnetfpn_pascal(...): Image segmentation on pascal voc with resnet-fpn.
semantic_segmentation(...): Semantic segmentation general.