ssd

Type Members

1. case class ComponetParam(nInput: Int, nPriors: Int, minSizes: Array[Float], maxSizes: Array[Float], aspectRatios: Array[Float], isFlip: Boolean, isClip: Boolean, variances: Array[Float], step: Int = 0, resolution: Int = 300) extends Product with Serializable

   

2. case class PreProcessParam(batchSize: Int = 4, resolution: Int = 300, pixelMeanRGB: (Float, Float, Float), hasLabel: Boolean, nPartition: Int, norms: (Float, Float, Float) = (1f, 1f, 1f)) extends Product with Serializable

   

3. class RoiImageToSSDBatch extends Preprocessing[ImageFeature, SSDMiniBatch]

   

   Convert a batch of labeled BGR images into a Mini-batch.

   Convert a batch of labeled BGR images into a Mini-batch.

   Notice: The totalBatch means a total batch size. In distributed environment, the batch should be divided by total core number

4. abstract class SSD[T] extends ImageModel[T]

   

   SSD model

5. class SSDMiniBatch extends MiniBatch[Float]

   

   A batch of data feed into the model.

   A batch of data feed into the model. The first size is batchsize

6. class SSDVGG[T] extends SSD[T]

   

   SSD model based on VGG16