Пример #1
0
        private unsafe List <YoloResult> Detect11(Mat mat, float thresh = 0.2f)
        {
            BoundingBox11 *elems;
            int            elemsSize;

            switch (_instanceId)
            {
            default:
            case 0:
                Wrapper0.Detect(mat.DataPointer, mat.Rows, mat.Cols, thresh, false, out elems, out elemsSize);
                break;

            case 1:
                Wrapper1.Detect(mat.DataPointer, mat.Rows, mat.Cols, thresh, false, out elems, out elemsSize);
                break;

            case 2:
                Wrapper2.Detect(mat.DataPointer, mat.Rows, mat.Cols, thresh, false, out elems, out elemsSize);
                break;

            case 3:
                Wrapper3.Detect(mat.DataPointer, mat.Rows, mat.Cols, thresh, false, out elems, out elemsSize);
                break;
            }
            return(GetResults(elems, elemsSize));
        }
Пример #2
0
        public Yolo(string cfgFile, string weightFile, int gpuId, int instanceId, int cudaVersion)
        {
            _instanceId  = instanceId;
            _cudaVersion = cudaVersion;

            switch (_instanceId)
            {
            default:
            case 0:
                Wrapper0.Init(cfgFile, weightFile, gpuId);
                break;

            case 1:
                Wrapper1.Init(cfgFile, weightFile, gpuId);
                break;

            case 2:
                Wrapper2.Init(cfgFile, weightFile, gpuId);
                break;

            case 3:
                Wrapper3.Init(cfgFile, weightFile, gpuId);
                break;
            }
        }
Пример #3
0
        public void Dispose()
        {
            switch (_instanceId)
            {
            default:
            case 0:
                Wrapper0.Close();
                break;

            case 1:
                Wrapper1.Close();
                break;

            case 2:
                Wrapper2.Close();
                break;

            case 3:
                Wrapper3.Close();
                break;
            }
        }