YoloV5 說明書

detect.py參數控制

--weights:模型權重的路徑或 Triton URL。
--source:要處理的輸入圖像或視頻的路徑、URL、glob、screen 或 webcam ID。
--data:(可選)數據集的 YAML 文件路徑。
--imgsz:推理時使用的輸入圖像大小,以高度和寬度為單位。
--conf-thres:置信度閾值,用於篩選預測結果中的邊界框。
--iou-thres:NMS(非極大值抑制)的 IoU(交並比)閾值。
--max-det:每張圖像最多檢測的目標數。
--device:指定要使用的計算設備,例如 cuda:0 或 cpu。
--view-img:是否在運行過程中顯示結果。
--save-txt:是否將預測結果保存到 .txt 文件中。
--save-conf:是否將置信度值保存到 .txt 標簽中。
--save-crop:是否將預測邊界框對應的圖像區域保存到獨立的圖像文件中。
--nosave:是否在運行過程中不保存結果圖像或視頻。
--classes:只顯示特定類別的預測結果,例如 --classes 0,或 --classes 0 2 3。
--agnostic-nms:是否使用類別不可知的 NMS。
--augment:是否使用增強推理(augmented inference)。
--visualize:是否可視化特徵。
--update:是否更新所有模型。
--project:指定結果保存的目錄。
--name:指定結果保存的文件名前綴。
--exist-ok:是否在結果保存目錄已存在的情況下覆蓋結果。
--line-thickness:邊界框線條的寬度(像素)。
--hide-labels:是否隱藏類別標簽。
--hide-conf:是否隱藏置信度值。
--half:是否使用 FP16 半精度推理。
--dnn:是否使用 OpenCV DNN 進行 ONNX 推理。
--vid-stride:處理視頻時跳過的幀數。