[deepstream][原创]过滤模型中某些类别

Posted 2021-12-11 FL1623863129

比如deepstream中yolov3是80类别，我只想要person这个类别，其他类别过率掉，怎么搞。这个其实很简单，修改插件源码即可

打开nvdsparsebbox_Yolo.cpp源码加个判断即可

static void addBBoxProposal(const float bx, const float by, const float bw, const float bh,

const uint stride, const uint& netW, const uint& netH, const int maxIndex,

const float maxProb, std::vector<NvDsInferParseObjectInfo>& binfo)

{

NvDsInferParseObjectInfo bbi = convertBBox(bx, by, bw, bh, stride, netW, netH);

if (bbi.width < 1 || bbi.height < 1) return;

bbi.detectionConfidence = maxProb;

bbi.classId = maxIndex;

if(bbi.classId==0)

binfo.push_back(bbi);

}

直接加一句if(bbi.classId==0)即可完成过滤任务！

结果展示：

参考文献：

1.Tried to use classId filtering with "infer-on-class-ids" into primary_gie but not working - DeepStream SDK - NVIDIA Developer Forums