diff --git a/README.md b/README.md index 8d9155d..335cb38 100644 --- a/README.md +++ b/README.md @@ -28,7 +28,7 @@ pip install opencv-python tqdm matplotlib cityscapesscripts yapf==0.31.0 2. 使用 `seg2city.py` 將 Roboflow 格式轉換為 Cityscapes 格式 3. 將轉換後的資料放至 `data/cityscapes/` -### 訓練與測試 +### 訓練與測試(需自己修改config檔案) ```bash # 訓練 @@ -55,6 +55,9 @@ python tools/pytorch2onnx_kneron.py \ --output-file work_dirs/kn_stdc1_in1k-pre_512x1024_80k_cityscapes/latest.onnx \ --verify +# onnx2nef +python onnx2nefSTDC630.py + # 將 NEF 複製到本機 docker cp :/data1/kneron_flow/models_630.nef \ "C:\Users\rd_de\stdc_git\work_dirs\nef\models_630.nef"