/dports/misc/py-mxnet/incubator-mxnet-1.9.0/scala-package/infer/src/test/scala/org/apache/mxnet/infer/ |
H A D | ClassifierSuite.scala | 113 val predictResult : IndexedSeq[Array[Float]] = constant 124 assertResult(predictResult(0).sortBy(-_)) { 147 assertResult(predictResult(0).sortBy(-_)) { 168 assertResult(predictResult(0)) { 190 assertResult(predictResult(0)) { 199 val predictResult: IndexedSeq[Array[Float]] = constant 212 assert(predictResult.size == result.size) 214 for(i <- predictResult.indices) { 215 assertResult(predictResult(i)) { 241 assert(predictResult.size == result.size) [all …]
|
H A D | ObjectDetectorSuite.scala | 91 val predictResult: IndexedSeq[IndexedSeq[(String, Array[Float])]] = constant 106 for (idx <- predictResult(0).indices) { 107 assert(predictResult(0)(idx)._1 == result(0)(idx)._1) 108 for (arridx <- predictResult(0)(idx)._2.indices) { 109 assert(predictResult(0)(idx)._2(arridx) == result(0)(idx)._2(arridx)) 141 val predictResult: IndexedSeq[IndexedSeq[(String, Array[Float])]] = constant 160 for (preidx <- predictResult.indices) { 161 for (idx <- predictResult(preidx).indices) { 162 assert(predictResult(preidx)(idx)._1 == result(preidx)(idx)._1) 163 for (arridx <- predictResult(preidx)(idx)._2.indices) { [all …]
|
H A D | PredictorSuite.scala | 73 val predictResult = IndexedSeq(NDArray.ones(Shape(1, 3, 2, 2))) constant 77 Mockito.doReturn(predictResult).when(testPredictor.mockModule) 101 val predictResult = IndexedSeq(NDArray.ones(Shape(1, 3, 2, 2), dtype = DType.Float64)) constant 105 Mockito.doReturn(predictResult).when(testPredictor.mockModule) 130 val predictResult = IndexedSeq(NDArray.ones(Shape(1, 3, 2, 2))) constant 134 Mockito.doReturn(predictResult).when(testPredictor.mockModule)
|
H A D | ImageClassifierSuite.scala | 105 val predictResult: IndexedSeq[IndexedSeq[(String, Float)]] = constant 110 predictResult(i).map(_._2).toArray
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/scala-package/infer/src/test/scala/org/apache/mxnet/infer/ |
H A D | ClassifierSuite.scala | 113 val predictResult : IndexedSeq[Array[Float]] = constant 124 assertResult(predictResult(0).sortBy(-_)) { 147 assertResult(predictResult(0).sortBy(-_)) { 168 assertResult(predictResult(0)) { 190 assertResult(predictResult(0)) { 199 val predictResult: IndexedSeq[Array[Float]] = constant 212 assert(predictResult.size == result.size) 214 for(i <- predictResult.indices) { 215 assertResult(predictResult(i)) { 241 assert(predictResult.size == result.size) [all …]
|
H A D | ObjectDetectorSuite.scala | 91 val predictResult: IndexedSeq[IndexedSeq[(String, Array[Float])]] = constant 106 for (idx <- predictResult(0).indices) { 107 assert(predictResult(0)(idx)._1 == result(0)(idx)._1) 108 for (arridx <- predictResult(0)(idx)._2.indices) { 109 assert(predictResult(0)(idx)._2(arridx) == result(0)(idx)._2(arridx)) 141 val predictResult: IndexedSeq[IndexedSeq[(String, Array[Float])]] = constant 160 for (preidx <- predictResult.indices) { 161 for (idx <- predictResult(preidx).indices) { 162 assert(predictResult(preidx)(idx)._1 == result(preidx)(idx)._1) 163 for (arridx <- predictResult(preidx)(idx)._2.indices) { [all …]
|
H A D | PredictorSuite.scala | 73 val predictResult = IndexedSeq(NDArray.ones(Shape(1, 3, 2, 2))) constant 77 Mockito.doReturn(predictResult).when(testPredictor.mockModule) 101 val predictResult = IndexedSeq(NDArray.ones(Shape(1, 3, 2, 2), dtype = DType.Float64)) constant 105 Mockito.doReturn(predictResult).when(testPredictor.mockModule) 130 val predictResult = IndexedSeq(NDArray.ones(Shape(1, 3, 2, 2))) constant 134 Mockito.doReturn(predictResult).when(testPredictor.mockModule)
|
H A D | ImageClassifierSuite.scala | 105 val predictResult: IndexedSeq[IndexedSeq[(String, Float)]] = constant 110 predictResult(i).map(_._2).toArray
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/scala-package/infer/src/main/scala/org/apache/mxnet/infer/ |
H A D | ObjectDetector.scala | 102 val predictResult = predictor.predictWithNDArray(input)(0).asInContext(Context.cpu()) constant 104 var batchResult = new ParArray[IndexedSeq[(String, Array[Float])]](predictResult.shape(0)) 105 (0 until predictResult.shape(0)).toArray.par.foreach( i => { 106 val r = predictResult.at(i) 110 handler.execute(predictResult.dispose()) 126 val predictResult = (0 until length).toArray.par.flatMap( i => { constant 140 var sortedIndices = predictResult.zipWithIndex.sortBy(-_._1(1)).map(_._2) 144 => (synset(predictResult(idx)(0).toInt), 145 predictResult(idx).takeRight(5))).toIndexedSeq 147 result = predictResult.map(ele
|
H A D | Classifier.scala | 109 val predictResult = predictor.predict(input)(0) constant 114 val sortedIndex = predictResult.zipWithIndex.sortBy(-_._1).map(_._2).take(topK.get) 115 result = sortedIndex.map(i => (synset(i), predictResult(i))).toIndexedSeq 117 result = synset.zip(predictResult).toIndexedSeq 148 val predictResult = predictResultPar.toArray constant 153 val sortedIndices = predictResult.map(r => 158 (synset(sIndx), predictResult(i)(sIndx))).toIndexedSeq 161 for (i <- predictResult.indices) { 162 result += synset.zip(predictResult(i)).toIndexedSeq
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/scala-package/infer/src/main/scala/org/apache/mxnet/infer/ |
H A D | ObjectDetector.scala | 102 val predictResult = predictor.predictWithNDArray(input)(0).asInContext(Context.cpu()) constant 104 var batchResult = new ParArray[IndexedSeq[(String, Array[Float])]](predictResult.shape(0)) 105 (0 until predictResult.shape(0)).toArray.par.foreach( i => { 106 val r = predictResult.at(i) 110 handler.execute(predictResult.dispose()) 126 val predictResult = (0 until length).toArray.par.flatMap( i => { constant 140 var sortedIndices = predictResult.zipWithIndex.sortBy(-_._1(1)).map(_._2) 144 => (synset(predictResult(idx)(0).toInt), 145 predictResult(idx).takeRight(5))).toIndexedSeq 147 result = predictResult.map(ele
|
H A D | Classifier.scala | 109 val predictResult = predictor.predict(input)(0) constant 114 val sortedIndex = predictResult.zipWithIndex.sortBy(-_._1).map(_._2).take(topK.get) 115 result = sortedIndex.map(i => (synset(i), predictResult(i))).toIndexedSeq 117 result = synset.zip(predictResult).toIndexedSeq 148 val predictResult = predictResultPar.toArray constant 153 val sortedIndices = predictResult.map(r => 158 (synset(sIndx), predictResult(i)(sIndx))).toIndexedSeq 161 for (i <- predictResult.indices) { 162 result += synset.zip(predictResult(i)).toIndexedSeq
|