Class ClassificationModel


  • public class ClassificationModel
    extends Model
    This class represents high-level API for classification models. ClassificationModel allows to set params for preprocessing input image. ClassificationModel creates net from file with trained weights and config, sets preprocessing input, runs forward pass and return top-1 prediction.
    • Constructor Detail

      • ClassificationModel

        protected ClassificationModel​(long addr)
      • ClassificationModel

        public ClassificationModel​(String model,
                                   String config)
        Create classification 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.
      • ClassificationModel

        public ClassificationModel​(String model)
        Create classification 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.
      • ClassificationModel

        public ClassificationModel​(Net network)
        Create model from deep learning network.
        Parameters:
        network - Net object.