yasenh / libtorch-yolov5

A LibTorch inference implementation of the yolov5
MIT License
372 stars 114 forks source link

Set different conf-thre for different tags #63

Open l0ngjx opened 1 year ago

l0ngjx commented 1 year ago

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);?