maskrcnn

Type Members

1. class MaskRCNN extends Container[Activity, Activity, Float]

   

2. case class MaskRCNNParams(anchorSizes: Array[Float] = Array[Float](32, 64, 128, 256, 512), aspectRatios: Array[Float] = Array[Float](0.5f, 1.0f, 2.0f), anchorStride: Array[Float] = Array[Float](4, 8, 16, 32, 64), preNmsTopNTest: Int = 1000, postNmsTopNTest: Int = 1000, preNmsTopNTrain: Int = 2000, postNmsTopNTrain: Int = 2000, rpnNmsThread: Float = 0.7f, minSize: Int = 0, boxResolution: Int = 7, maskResolution: Int = 14, scales: Array[Float] = ..., samplingRatio: Int = 2, boxScoreThresh: Float = 0.05f, boxNmsThread: Float = 0.5f, maxPerImage: Int = 100, outputSize: Int = 1024, layers: Array[Int] = Array[Int](256, 256, 256, 256), dilation: Int = 1, useGn: Boolean = false) extends Product with Serializable