Skip to content
Snippets Groups Projects
Commit 6a3e1505 authored by anzhengqi's avatar anzhengqi
Browse files

add posenet export air

parent 07f7db59
No related branches found
No related tags found
No related merge requests found
......@@ -285,13 +285,13 @@ PoseNet是剑桥大学提出的一种鲁棒、实时的6DOF(单目六自由度
### 导出
在执行推理之前,需要通过export.py导出mindir文件
在执行推理之前,需要通过export.py导出MINDIR/AIR模型
```shell
python export.py --ckpt_url [CKPT_URL] --dataset [DATASET]
python export.py --ckpt_url [CKPT_URL] --dataset [DATASET] --file_format [FILE_FORMAT]
```
ckpt_url和dataset为必填项。
CKPT_URL和DATASET为必填项, FILE_FORMAT默认为MINDIR,可配值MINDIR或AIR.
## 推理过程
......
......@@ -34,7 +34,7 @@ parser.add_argument('--dataset', type=str, default='KingsCollege',
help='Name of dataset.')
parser.add_argument("--ckpt_url", type=str, required=True, help="Checkpoint file path.")
parser.add_argument("--file_name", type=str, default="posenet", help="output file name.")
parser.add_argument('--file_format', type=str, choices=["MINDIR"], default='MINDIR', help='file format')
parser.add_argument('--file_format', type=str, choices=["MINDIR", "AIR"], default='MINDIR', help='file format')
args = parser.parse_args()
context.set_context(mode=context.GRAPH_MODE, device_target=args.device_target)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment