breezedeus/cnocr

cnocr训练镜像文档

镜像概述

本镜像为cnocr（中文OCR工具包）提供训练环境，基于Python 3.6.10构建，包含运行和训练cnocr所需的全套依赖库，可直接用于OCR模型训练、推理及相关应用开发。

核心功能和特性

• 基础环境：Python 3.6.10运行环境，确保与cnocr兼容性
• 深度学习框架：集成mxnet-cu80 1.4.1，支持GPU加速训练
• 计算机视觉库：包含opencv-python 4.1.2.30，支持图像处理与预处理
• 科学计算工具：集成numpy 1.14.6、scipy 1.2.1，提供数值计算支持
• 机器学习库：包含scikit-learn 0.21.0，支持数据预处理与模型评估
• 可视化工具：集成matplotlib 3.0.3，支持训练过程可视化
• 交互开发环境：包含jupyterlab 0.35.6，支持交互式开发与调试

依赖包详情

主要依赖包及版本：

• mxnet-cu80 1.4.1（深度学习框架，支持CUDA 8.0）
• gluoncv 0.4.0.post0（计算机视觉工具库）
• gluonnlp 0.6.0（自然语言处理工具库）
• opencv-python 4.1.2.30（计算机视觉处理）
• numpy 1.14.6（数值计算基础库）
• scipy 1.2.1（科学计算库）
• scikit-learn 0.21.0（机器学习库）
• matplotlib 3.0.3（数据可视化库）
• jupyterlab 0.35.6（交互式开发环境）

使用场景

• 中文及多语言文本识别模型的训练与调优
• OCR相关应用的快速原型开发
• 教育和研究领域的OCR技术学习与实验
• 批量文本识别任务的处理与部署测试

使用方法

基本使用

拉取并运行镜像（假设镜像名称为cnocr-training）：

bash
docker run -it --rm cnocr-training

启动JupyterLab

运行容器并启动JupyterLab服务：

bash
docker run -it -p 8888:8888 --rm cnocr-training jupyter lab --ip=0.0.0.0 --allow-root

启动后，通过浏览器访问http://localhost:8888，输入容器中显示的token即可进入JupyterLab界面。

挂载数据卷

挂载本地数据集和代码目录，便于训练：

bash
docker run -it -v /本地数据目录:/data -v /本地代码目录:/code --rm cnocr-training

训练示例

在容器内执行cnocr训练命令（具体命令需参考cnocr官方文档）：

bash
# 假设已准备好训练数据
python -m cnocr train --train_data /data/train_data --epochs 10

注意事项

• 本镜像基于CUDA 8.0构建，需在支持CUDA的环境中使用GPU加速
• 首次运行JupyterLab时，需记录终端输出的访问token
• 大规模训练建议挂载外部存储，避免容器内存储空间不足
• 如需更新依赖包，可在容器内使用pip install --upgrade 包名命令