文档中心

打包镜像

训练获得模型文件后，需与代码一同打包为镜像，通过镜像启动在线推理服务（关于在线推理服务的代码结构参阅训练代码简介）。下载Retrain推理代码包：

并将路径及其所有子路径和文件保存在本地（此处默认保存在/data/目录下）。将frozen_inference_graph.pb文件和label_map.txt（见：模型训练）保存在其子路径：/data/retrain/code/checkpoint_dir/中。本地路径中保存的所有文件为：

/data/retrain/[路径]
/data/retrain/retrain-detect-cpu.Dockerfile
/data/retrain/retrain-detect.conf
/data/retrain/code/[路径]
/data/retrain/code/retrain_inference.py
/data/retrain/code/retrain_conf.py
/data/retrain/code/checkpoint_dir/[路径]
/data/retrain/code/checkpoint_dir/frozen_inference_graph.pb
/data/retrain/code/checkpoint_dir/label_map.pbtxt

其中/data/retrain/为根目录，其下保存了Dockerfile和conf文件用于打包镜像，以及code子目录。code中保存了代码（共计2个.py文件）和checkpoint_dir子目录。将之前生成的pb模型文件和label_map.pbtxt物体类别标签文件放在/data/object-detect/code/checkpoint_dir/目录下。文件已准备好打包。

转至根目录/data/retrain/，运行命令打包镜像，标签为：uhub.service.auto-ai.com.cn//retrain-detect-infer:test，引用容器文件：retrain-detect-cpu.Dockerfile：

sudo docker build -t uhub.auto-ai.com.cn//retrain-detect-infer:test -f retrain-detect-cpu.Dockerfile .

请注意命令的最后有参数为单个英文句号“.”。等待打包完毕。完成后，获得镜像：uhub.service.auto-ai.com.cn//retrain-detect-infer:test。在本地主机登录UHub容器管理：

sudo docker login uhub.auto-ai.com.cn

分别输入云知芯管理账号和密码，等待login success指示登录完毕。输入命令将打包好的镜像上传至Uhub镜像库：

sudo docker push uhub.auto-ai.com.cn//retrain-detect-infer:test

等待上传完毕。镜像已经上传至镜像库，参阅启动在线服务使用该镜像进行推理。

更多关于镜像打包的信息参阅使用自定义镜像打包