libtorch-yolov5 icon indicating copy to clipboard operation
libtorch-yolov5 copied to clipboard

Published 20 hours ago • verified publisher icon yasenh

Set different conf-thre for different tags

Open l0ngjx opened this issue 2 years ago • 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);？

l0ngjx avatar Sep 19 '22 04:09 l0ngjx