Override this method to define a model.
Override this method to define a model.
Evaluate the ImageSet given validation methods.
Evaluate the ImageSet given validation methods. Currently only DistributedImageSet is supported.
DistributedImageSet in which each image should have a label.
Array of ValidationMethod to evaluate the ImageSet.
An instance of ImageConfigure. Default is null and it will be pre-defined together with each ImageModel.
The defined model, either from buildModel() or loaded from file.
The defined model, either from buildModel() or loaded from file.
Predict for classes.
Predict for classes. By default, label predictions start from 0.
Prediction data, RDD of Sample.
Number of samples per batch. Default is 32.
Boolean. Whether result labels start from 0. Default is true. If false, result labels start from 1.
Computes the output using the current parameter set of the class and input.
Computes the output using the current parameter set of the class and input. This function returns the result which is stored in the output field.
Save the model to the specified path.
Save the model to the specified path.
The path to save the model. Local file system, HDFS and Amazon S3 are supported. HDFS path should be like "hdfs://[host]:[port]/xxx". Amazon S3 path should be like "s3a://bucket/xxx".
The path to save weights. Default is null.
Whether to overwrite the file if it already exists. Default is false.
Set the model to be in evaluate status, i.e.
Set the model to be in evaluate status, i.e. remove the effect of Dropout, etc.
Print out the summary of the model.
Print out the summary of the model.
(Since version 0.3.0) please use recommended saveModule(path, overWrite)
SSD model