模型推論與自動匯出

主推論頁面支援範圍

“模型推論驗證”支援四種現代任務,以及經典目標偵測和經典分類。異常偵測請使用異常偵測自己的推論頁。

架構 輸入權重 還需要什麼
現代架構 .pt 或 .onnx 匹配的任務型別;ONNX 尤其需要選對任務
經典偵測/分類 .pdparams 與模型匹配的訓練 YAML,可從訓練實驗匯入
異常偵測自己的頁面 model.patchcore.pt 同頁的圖片來源、裝置及閾值倍率

經典訓練實驗匯入會優先選擇有效的最終配置快照和實際檢查點,不用 YAML 中的預訓練 URL 冒充已經訓練好的模型。必要時可指定類別 TXT;經典權重形狀不匹配會報錯,不允許悄悄使用隨機參數。

任務型別、閾值與裝置

現代任務包括分類、目標偵測、實例分割和旋轉框偵測。切換為經典架構後任務型別由所選架構固定。

置信度閾值介面預設為 0.3,可輸入 0.01~1.0。偵測類推論按閾值保留結果;經典分類返回達到閾值的最多前 5 個類別。現代分類的原生結果處理不按這個 conf 控制項過濾分類機率,不能把該控制項解釋為現代分類的 OK/NG 判定規則。

現代 PT 支援按裝置選項執行;當前軟體將現代 ONNX 強制回退 CPU 推論,即使選了 GPU,也會在日誌中提示,不承諾 ONNX GPU 推論。經典推論可在 CPU 或所選的一張 NVIDIA GPU 執行。

前處理

軟體生成的現代分類 PT/ONNX 內含中心裁剪前處理後設資料,軟體推論會讀取;只有 ONNX 外部推論程式不會因為旁邊存在 JSON 就自動應用前處理,仍需部署方正確實現。

經典分類從配置的 Infer 前處理和類別對映讀取;經典偵測使用匹配的 TestReader。任意外部權重若缺少對應訓練配置,僅靠副檔名不能還原正確前處理。

輸出目錄

現代推論不指定輸出目錄時由模型任務決定預設位置;顯式指定目錄時直接儲存到該目錄,允許複用,注意同名結果可能被覆蓋。

經典和異常偵測推論在輸出根目錄下新建獨立實驗目錄,儲存視覺化及結構化結果。實際輸出路徑以完成日誌為準。

訓練自動匯出

訓練架構 成功後匯出
四種現代架構 ONNX;偵測/分割/OBB 另有 infer_cfg.yml,分類另有 Label.txt 與前處理資訊
經典偵測 model.pdmodel、model.pdiparams、參數資訊檔案及 infer_cfg.yml
經典分類 推論模型、參數、ImageSize.txt、部署 YAML、類別清單
異常偵測 自包含 model.patchcore.pt,使用異常偵測推論流程

現代 ONNX 預設不簡化圖,不自動安裝 onnxslim;OBB 不內嵌 NMS。經典匯出在本次獨立程序內使用 CPU,訓練可仍為 GPU。軟體在訓練完成後可能繼續顯示“正在匯出”,要等匯出結果明確後再拿模型部署。

訓練完成不代表匯出一定完成。經典匯出失敗會保留訓練權重和完整實驗檔案,可檢視 export.json 與原始日誌;不要刪除整個實驗來“重試匯出”。