- Mindspore=2.0.0
- HRNet
克隆该仓库:
git clone https://github.com/Calylyli/Mindsporehrnet.git
- 准备数据集 (cityscapes).
├─ cityscapes # cityscapes数据集根目录
│ ├─ gtFine # 标签文件
│ │ ├─ train # 训练标签文件
│ │ │ └─ [city folders]
│ │ │ └─ [label images]
│ │ └─ val # 推理标签文件
│ │ └─ [city folders]
│ │ └─ [label images]
│ ├─ leftImg8bit # 图像文件
│ │ ├─ train # 训练图像文件
│ │ │ └─ [city folders]
│ │ │ └─ [images]
│ │ └─ val # 推理图像文件
│ │ └─ [city folders]
│ │ └─ [images]
│ ├─ train.lst # 训练样本列表
│ └─ val.lst # 推理样本列表
- 运行如下代码 :
python train.py
- 运行如下代码 :
python export.py --device_id [DEVICE_ID] --checkpoint_file [CKPT_PATH] --file_name [FILE_NAME] --file_format MINDIR --device_target Ascend --dataset [DATASET]
- 运行如下代码 :
sh scripts/run_eval.sh [DEVICE_ID] [DATASET_PATH] [CHECKPOINT_PATH]
- 运行如下代码 :
sh scripts/ascend310_inference.sh [MINDIR_PATH] [DATA_PATH] [DEVICE_ID]