25 lines
636 B
YAML
25 lines
636 B
YAML
# 模型類型
|
||
model_type: onnx
|
||
|
||
# NEF 模型路徑（確保這是你的最終模型）
|
||
model_path: "/data1/kneron_flow/models_720.nef"
|
||
|
||
# YOLOv5 輸入尺寸
|
||
input_w: 640
|
||
input_h: 640
|
||
|
||
# 類別數量（你的模型是 6 類）
|
||
nc: 6
|
||
|
||
# 信心閾值（影響檢測結果數量）
|
||
conf_thres: 0.3
|
||
iou_thres: 0.5
|
||
|
||
# 是否使用 COCO 格式輸出（如果你的模型不是 COCO 格式，請設為 False）
|
||
e2e_coco: False
|
||
|
||
# Anchor Grid `.npy` 檔案路徑（確保它們存在）
|
||
grid20_path: "/workspace/yolov5/npy/20_640x640.npy"
|
||
grid40_path: "/workspace/yolov5/npy/40_640x640.npy"
|
||
grid80_path: "/workspace/yolov5/npy/80_640x640.npy"
|