YOLO训练产出warning: NMS time limit 1.060s exceeded原因与解决办法

慈云数据 8个月前 (03-12) 技术支持 131 0

在进行模型训练结束后,模型代码会执行

Python val.py

对模型进行map准确率的验证使用时候出现

在这里插入图片描述

talk is Cheap ,show me the code.

找到warning的代码出处:

def non_max_suppression(prediction, conf_thres=0.25, iou_thres=0.45, classes=None, agnostic=False, multi_label=False,
                        labels=(), max_det=300):
    """Runs Non-Maximum Suppression (NMS) on inference results
    Returns:
         list of detections, on (n,6) tensor per image [xyxy, conf, cls]
    """
    nc = prediction.shape[2] - 5  # number of classes
    xc = prediction[..., 4] >; conf_thres  # candidates
    # Checks
    assert 0 
微信扫一扫加客服

微信扫一扫加客服

点击启动AI问答
Draggable Icon