更新 README.md
This commit is contained in:
parent
fbd65df831
commit
bb177d286a
@ -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 <container_id>:/data1/kneron_flow/models_630.nef \
|
||||
"C:\Users\rd_de\stdc_git\work_dirs\nef\models_630.nef"
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user