Class SegmentationModel


  • public class SegmentationModel
    extends Model
    This class represents high-level API for segmentation models SegmentationModel allows to set params for preprocessing input image. SegmentationModel creates net from file with trained weights and config, sets preprocessing input, runs forward pass and returns the class prediction for each pixel.
    • Constructor Detail

      • SegmentationModel

        protected SegmentationModel​(long addr)
      • SegmentationModel

        public SegmentationModel​(java.lang.String model,
                                 java.lang.String config)
        Create segmentation model from network represented in one of the supported formats. An order of model and config arguments does not matter.
        Parameters:
        model - Binary file contains trained weights.
        config - Text file contains network configuration.
      • SegmentationModel

        public SegmentationModel​(java.lang.String model)
        Create segmentation model from network represented in one of the supported formats. An order of model and config arguments does not matter.
        Parameters:
        model - Binary file contains trained weights.
      • SegmentationModel

        public SegmentationModel​(Net network)
        Create model from deep learning network.
        Parameters:
        network - Net object.
    • Method Detail

      • segment

        public void segment​(Mat frame,
                            Mat mask)
        Given the input frame, create input blob, run net
        Parameters:
        mask - Allocated class prediction for each pixel
        frame - automatically generated
      • finalize

        protected void finalize()
                         throws java.lang.Throwable
        Overrides:
        finalize in class Model
        Throws:
        java.lang.Throwable