yolov5_config.py 564 Bytes
from easydict import EasyDict as edict

config = edict(
    # weights='/home/situ/qfs/invoice_tamper/09_project/project/yolov5_inference/runs/exp2/weights/best.pt',  # model path or triton URL
    weights='runs/train/exp/weights/best.onnx',  # model path or triton URL
    data='data/VOC.yaml',  # dataset.yaml path
    imgsz=(640, 640),  # inference size (height, width)
    conf_thres=0.2,  # 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
)