コード例 #1
0
 /// <inheritdoc/>
 Tensor IOps.NonMaxSuppression(Tensor[] tensors, int maxOutputBoxesPerClass, float iouThreshold, float scoreThreshold, int centerPointBox)
 {
     var O = m_Ops.NonMaxSuppression(tensors, maxOutputBoxesPerClass, iouThreshold, scoreThreshold, centerPointBox);
     LogLayerSummary($"{string.Join(",", Enumerable.Select(tensors, t => t.shape.ToString()))} centerPointBox: {centerPointBox} # {O.shape}");
     LogOutputTensorSummary(O, Prefix + nameof(IOps.NonMaxSuppression));
     return O;
 }
コード例 #2
0
ファイル: StatsOps.cs プロジェクト: amaannt/PracticeMLAgents
        /// <inheritdoc/>
        Tensor IOps.NonMaxSuppression(Tensor[] tensors, int maxOutputBoxesPerClass, float iouThreshold, float scoreThreshold, int centerPointBox)
        {
            var O = m_Ops.NonMaxSuppression(tensors, maxOutputBoxesPerClass, iouThreshold, scoreThreshold, centerPointBox);

            // @TODO: not implemented
            return(O);
        }
コード例 #3
0
        /// <inheritdoc/>
        Tensor IOps.NonMaxSuppression(Tensor[] tensors, int maxOutputBoxesPerClass, float iouThreshold, float scoreThreshold, int centerPointBox)
        {
            var Y = m_Ops1.NonMaxSuppression(tensors, maxOutputBoxesPerClass, iouThreshold, scoreThreshold, centerPointBox);
            var Z = m_Ops2.NonMaxSuppression(tensors, maxOutputBoxesPerClass, iouThreshold, scoreThreshold, centerPointBox);

            CheckSame(Y, Z, Layer.Type.NonMaxSuppression);
            return(Y);
        }
コード例 #4
0
        /// <inheritdoc/>
        Tensor IOps.NonMaxSuppression(Tensor[] tensors, int maxOutputBoxesPerClass, float iouThreshold, float scoreThreshold, int centerPointBox)
        {
            var O = m_Ops.NonMaxSuppression(tensors, maxOutputBoxesPerClass, iouThreshold, scoreThreshold, centerPointBox);

            D.Log($"{string.Join(",", tensors.Select(t => t.shape.ToString()))} centerPointBox: {centerPointBox} # {O.shape}");
            O.PrintDataPart(32, Prefix + nameof(IOps.NonMaxSuppression));
            return(O);
        }
コード例 #5
0
        /// <inheritdoc/>
        Tensor IOps.NonMaxSuppression(Tensor[] tensors, int maxOutputBoxesPerClass, float iouThreshold, float scoreThreshold, int centerPointBox)
        {
            var O = m_Ops.NonMaxSuppression(tensors, maxOutputBoxesPerClass, iouThreshold, scoreThreshold, centerPointBox);

            m_Alu += 0;
            m_Mem += 0;
            RegisterLayerStats();
            return(O);
        }