diff --git a/official/cv/posenet/README_CN.md b/official/cv/posenet/README_CN.md index a99073c2a7e590127f01edb7d1edbcfb368ff121..27d50f7490f858a1a366aad62f514cb60e1e12ac 100644 --- a/official/cv/posenet/README_CN.md +++ b/official/cv/posenet/README_CN.md @@ -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鍜宒ataset涓哄繀濉」銆� +CKPT_URL鍜孌ATASET涓哄繀濉」, FILE_FORMAT榛樿涓篗INDIR锛屽彲閰嶅€糓INDIR鎴朅IR. ## 鎺ㄧ悊杩囩▼ diff --git a/official/cv/posenet/export.py b/official/cv/posenet/export.py index 9364189f827c08aca75980e31a53f6463c192423..4166e946e32dc4f46336b60a90b5b1f242c9d531 100644 --- a/official/cv/posenet/export.py +++ b/official/cv/posenet/export.py @@ -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)