Slim

1/2ページ

slim ckpt檔案轉pb檔案

2. 模型固化 2.1 模型固化小工具freeze_graph.py 實際上使用訓練得到模型時,需要把權重固定,不然每次測試一張圖片就相當於繼續訓練模型,權重也要重新變化一次,這會導致測試一張圖片需要很久的時間。在固定權重的基礎上,還需要整合圖的定義。這個過程就是 freeze graph 。 這裡 […]