yolov5_config.py 425 Bytes
from easydict import EasyDict as edict

config = edict(
    weights='runs/train/exp/weights/best.pt',  # model path or triton URL
    data='data/VOC.yaml',  # dataset.yaml path
    imgsz=(640, 640),  # inference size (height, width)
    conf_thres=0.5,  # confidence threshold
    iou_thres=0.45,  # NMS IOU threshold
    max_det=1000,  # maximum detections per image
    device=''  # cuda device, i.e. 0 or 0,1,2,3 or cpu
)