yasenh/libtorch-yolov5

Set different conf-thre for different tags

l0ngjx opened this issue · 0 comments

If I want to set different conf-thre for different tags, what should I do?
如果我想要给不同标签设置不同conf-thre,我应该怎么做呢?
auto conf_mask = detections.select(2, 4).ge(conf_thres).unsqueeze(2);
or cv::dnn::NMSBoxes(offset_box_vec, score_vec, conf_thres, iou_thres, nms_indices);?