更新 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 格式
|
2. 使用 `seg2city.py` 將 Roboflow 格式轉換為 Cityscapes 格式
|
||||||
3. 將轉換後的資料放至 `data/cityscapes/`
|
3. 將轉換後的資料放至 `data/cityscapes/`
|
||||||
|
|
||||||
### 訓練與測試
|
### 訓練與測試(需自己修改config檔案)
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
# 訓練
|
# 訓練
|
||||||
@ -55,6 +55,9 @@ python tools/pytorch2onnx_kneron.py \
|
|||||||
--output-file work_dirs/kn_stdc1_in1k-pre_512x1024_80k_cityscapes/latest.onnx \
|
--output-file work_dirs/kn_stdc1_in1k-pre_512x1024_80k_cityscapes/latest.onnx \
|
||||||
--verify
|
--verify
|
||||||
|
|
||||||
|
# onnx2nef
|
||||||
|
python onnx2nefSTDC630.py
|
||||||
|
|
||||||
# 將 NEF 複製到本機
|
# 將 NEF 複製到本機
|
||||||
docker cp <container_id>:/data1/kneron_flow/models_630.nef \
|
docker cp <container_id>:/data1/kneron_flow/models_630.nef \
|
||||||
"C:\Users\rd_de\stdc_git\work_dirs\nef\models_630.nef"
|
"C:\Users\rd_de\stdc_git\work_dirs\nef\models_630.nef"
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user