集装箱号识别项目

集装箱号识别项目

项目地址：集装箱识别项目

该项目基于 mmOCR 0.2.0

主分支目前支持 PyTorch 1.5 以上的版本。

文本检测算法

PANet (ICCV’2019)

文字识别算法

RobustScanner (ECCV’2020)

使用教程

构建数据集

1. 检查集装箱号标记
2. 数据增强
3. 切下文本区域构建识别数据集

python tools/data/create_dataset.py

4. 构建文本检测数据集

python tools/data/textdet/icdar_converter.py

训练检测模型

python tools/data/create_dataset.py --config "configs/textdet/fcenet/fcenet_r50dcnv2_fpn_1500e_ctw1500.py" --load-from "checkpoints/
                fcenet_r50dcnv2_fpn_1500e_ctw1500-05d740bb.pth"

训练识别模型

python tools/data/create_dataset.py --config "configs/textrecog/robust_scanner/robustscanner_r31_academic.py" --load-from "checkpoints/robustscanner_r31_academic-5f05874f.pth"

测试集装箱的测试结果（图片）

python container/ocr_image_demo.py

测试集装箱的测试结果（视频）

python container/getframes.py  # 读取视频
python listen_path.py  # 预测集装箱号

网页展示结果

sh web/start.sh

开源许可证

该项目采用 Apache 2.0 license 开源许可证。

引用

如果您发现此项目对您的研究有用，请考虑引用：

@misc{container-code-recognition2021,
    title={container-code-recognition2021},
    author={students},
    howpublished = {\url{https://github.com/liuweijie19980216/container-code-recognition}},
    year={2021}
}

更新日志

初始版本发布于2021.8.11

致谢

感谢优秀的开源项目MMOCR和一起参与贡献的同学们。