|
|
|
@ -414,7 +414,7 @@ bool YoloV5NcnnDetect( ncnn::Mat& mat, bool use_gpu, std::vector<IDevice::RECOG_
|
|
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
bool YoloV5NcnnDetect( cv::Mat& mat, bool use_gpu, std::vector<IDevice::RECOG_OBJECT>& objects)
|
|
|
|
|
bool YoloV5NcnnDetect( cv::Mat& mat, bool use_gpu, const std::string& blobName8, const std::string& blobName16, const std::string& blobName32, std::vector<IDevice::RECOG_OBJECT>& objects)
|
|
|
|
|
{
|
|
|
|
|
if (use_gpu && ncnn::get_gpu_count() == 0)
|
|
|
|
|
{
|
|
|
|
@ -480,7 +480,7 @@ bool YoloV5NcnnDetect( cv::Mat& mat, bool use_gpu, std::vector<IDevice::RECOG_OB
|
|
|
|
|
// stride 8
|
|
|
|
|
{
|
|
|
|
|
ncnn::Mat out;
|
|
|
|
|
ex.extract("output", out);
|
|
|
|
|
ex.extract(blobName8.c_str(), out);
|
|
|
|
|
|
|
|
|
|
ncnn::Mat anchors(6);
|
|
|
|
|
anchors[0] = 10.f;
|
|
|
|
@ -499,7 +499,7 @@ bool YoloV5NcnnDetect( cv::Mat& mat, bool use_gpu, std::vector<IDevice::RECOG_OB
|
|
|
|
|
// stride 16
|
|
|
|
|
{
|
|
|
|
|
ncnn::Mat out;
|
|
|
|
|
ex.extract("781", out);
|
|
|
|
|
ex.extract(blobName16.c_str(), out);
|
|
|
|
|
|
|
|
|
|
ncnn::Mat anchors(6);
|
|
|
|
|
anchors[0] = 30.f;
|
|
|
|
@ -518,7 +518,7 @@ bool YoloV5NcnnDetect( cv::Mat& mat, bool use_gpu, std::vector<IDevice::RECOG_OB
|
|
|
|
|
// stride 32
|
|
|
|
|
{
|
|
|
|
|
ncnn::Mat out;
|
|
|
|
|
ex.extract("801", out);
|
|
|
|
|
ex.extract(blobName32.c_str(), out);
|
|
|
|
|
|
|
|
|
|
ncnn::Mat anchors(6);
|
|
|
|
|
anchors[0] = 116.f;
|
|
|
|
|