案例4—数字识别系统
1. 环境安装
1.1 下载代码
下载源码压缩包:digit_demo.zip
digit_demo.zip 仅支持识别数字和字母,若需识别文字,使用 ocr_demo.zip。
解压命令:
unzip digit_demo.zip -d ~/
1.2 安装系统依赖
sudo apt update
sudo apt install python3-virtualenv
1.3 安装Python依赖
(1)创建Python虚拟环境
python3 -m venv ~/ocr-venv
(2)配置 pip 源为进迭时空镜像源
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
pip config set global.extra-index-url
https://git.spacemit.com/api/v4/projects/33/packages/pypi/simple
(3)安装项目依赖
cd ~/digit_demo
source ~/ocr-venv/bin/activate
pip install -r requirements.txt
2. 程序启动
- 进入项目目录并激活虚拟环境
cd ~/digit_demo
source ~/ocr-venv/bin/activate
2.1 OCR识别
- OCR识别:调用检测+识别模型进行推理,仅识别数字和英文
python ocr_demo.py --mode camera #启用摄像头推理
python ocr_demo.py --mode image --image_path(默认为spacemit_orc/data/train_img_33.jpg)
- OCR识别:仅调用识别模型进行推理,仅识别数字和英文,识别速度更快
python test_recognize.py --use_camera #启用摄像头推理
2.2 数码管显示
python digitial_serial.py
2.3 OCR识别并驱动数码管显示
python recognize_and_dispaly.py --use_camera # 摄像头识别