Skip to content

P
PaddleOCR

weixin_41840029 / PaddleOCR
与 Fork 源项目一致

Fork自 PaddlePaddle / PaddleOCR

通知 1
Star 1
Fork 0
P
PaddleOCR
未验证 提交 91f5ab5c 编写于 作者: M MissPenguin 提交者： GitHub
浏览文件
操作

Merge pull request #1365 from WenmuZhou/tree_doc 

[Dygraph] add tree doc
上级 5b2f6b75 25b8b35c
显示空白变更内容
内联 并排
Showing with 341 addition and 322 deletion
doc/doc_ch/tree.md
浏览文件 @ 91f5ab5c
...@@ -4,18 +4,14 @@ PaddleOCR 的整体目录结构介绍如下： ...@@ -4,18 +4,14 @@ PaddleOCR 的整体目录结构介绍如下：
``` ```
PaddleOCR PaddleOCR
├── configs // 配置文件，可通过yml文件选择模型结构并修改超参 ├── configs // 配置文件，可通过 yml 文件选择模型结构并修改超参
│ ├── cls // 方向分类器相关配置文件 │ ├── cls // 方向分类器相关配置文件
│ │ ├── cls_mv3.yml // 训练配置相关，包括骨干网络、head、loss、优化器 │ │ ├── cls_mv3.yml // 训练配置相关，包括骨干网络、head、loss、优化器和数据
│ │ └── cls_reader.yml // 数据读取相关，数据读取方式、数据存储路径
│ ├── det // 检测相关配置文件 │ ├── det // 检测相关配置文件
│ │ ├── det_db_icdar15_reader.yml // 数据读取
│ │ ├── det_mv3_db.yml // 训练配置 │ │ ├── det_mv3_db.yml // 训练配置
│ │ ... │ │ ...
│ └── rec // 识别相关配置文件 │ └── rec // 识别相关配置文件
│ ├── rec_benchmark_reader.yml // LMDB 格式数据读取相关 │ ├── rec_mv3_none_bilstm_ctc.yml // crnn 训练配置
│ ├── rec_chinese_common_train.yml // 通用中文训练配置
│ ├── rec_icdar15_reader.yml // simple 数据读取相关，包括数据读取函数、数据路径、标签文件
│ ... │ ...
├── deploy // 部署相关 ├── deploy // 部署相关
│ ├── android_demo // android_demo │ ├── android_demo // android_demo
...@@ -59,6 +55,11 @@ PaddleOCR ...@@ -59,6 +55,11 @@ PaddleOCR
│ │ ├── README.md │ │ ├── README.md
│ │ └── sample_request.txt │ │ └── sample_request.txt
│ ├── hubserving // hubserving │ ├── hubserving // hubserving
│ │ ├── ocr_cls // 方向分类器
│ │ │ ├── config.json // serving 配置
│ │ │ ├── __init__.py
│ │ │ ├── module.py // 预测模型
│ │ │ └── params.py // 预测参数
│ │ ├── ocr_det // 文字检测 │ │ ├── ocr_det // 文字检测
│ │ │ ├── config.json // serving 配置 │ │ │ ├── config.json // serving 配置
│ │ │ ├── __init__.py │ │ │ ├── __init__.py
...@@ -83,9 +84,9 @@ PaddleOCR ...@@ -83,9 +84,9 @@ PaddleOCR
│ │ ├── cls_process.cc // 方向分类器数据处理 │ │ ├── cls_process.cc // 方向分类器数据处理
│ │ ├── cls_process.h │ │ ├── cls_process.h
│ │ ├── config.txt // 检测配置参数 │ │ ├── config.txt // 检测配置参数
│ │ ├── crnn_process.cc // crnn数据处理 │ │ ├── crnn_process.cc // crnn 数据处理
│ │ ├── crnn_process.h │ │ ├── crnn_process.h
│ │ ├── db_post_process.cc // db数据处理 │ │ ├── db_post_process.cc // db 数据处理
│ │ ├── db_post_process.h │ │ ├── db_post_process.h
│ │ ├── Makefile // 编译文件 │ │ ├── Makefile // 编译文件
│ │ ├── ocr_db_crnn.cc // 串联预测 │ │ ├── ocr_db_crnn.cc // 串联预测
...@@ -108,101 +109,108 @@ PaddleOCR ...@@ -108,101 +109,108 @@ PaddleOCR
│ └── README.md // 说明文档 │ └── README.md // 说明文档
├── doc // 文档教程 ├── doc // 文档教程
│ ... │ ...
├── paddleocr.py
├── ppocr // 网络核心代码 ├── ppocr // 网络核心代码
│ ├── data // 数据处理 │ ├── data // 数据处理
│ │ ├── cls // 方向分类器 │ │ ├── imaug // 图片和 label 处理代码
│ │ │ ├── dataset_traversal.py // 数据传输，定义数据读取器，读取数据并组成batch │ │ │ ├── text_image_aug // 文本识别的 tia 数据扩充
│ │ │ └── randaugment.py // 随机数据增广操作 │ │ │ │ ├── __init__.py
│ │ ├── det // 检测 │ │ │ │ ├── augment.py // tia_distort,tia_stretch 和 tia_perspective 的代码
│ │ │ ├── data_augment.py // 数据增广操作 │ │ │ │ ├── warp_mls.py
│ │ │ ├── dataset_traversal.py // 数据传输，定义数据读取器，读取数据并组成batch │ │ │ ├── __init__.py
│ │ │ ├── db_process.py // db 数据处理 │ │ │ ├── east_process.py // EAST 算法的数据处理步骤
│ │ │ ├── east_process.py // east 数据处理
│ │ │ ├── make_border_map.py // 生成边界图 │ │ │ ├── make_border_map.py // 生成边界图
│ │ │ ├── make_shrink_map.py // 生成收缩图 │ │ │ ├── make_shrink_map.py // 生成收缩图
│ │ │ ├── random_crop_data.py // 随机切割 │ │ │ ├── operators.py // 图像基本操作，如读取和归一化
│ │ │ └── sast_process.py // sast 数据处理 │ │ │ ├── randaugment.py // 随机数据增广操作
│ │ ├── reader_main.py // 数据读取器主函数 │ │ │ ├── random_crop_data.py // 随机裁剪
│ │ └── rec // 识别 │ │ │ ├── rec_img_aug.py // 文本识别的数据扩充
│ │ ├── dataset_traversal.py // 数据传输，定义数据读取器，包含 LMDB_Reader 和 Simple_Reader │ │ │ └── sast_process.py // SAST 算法的数据处理步骤
│ │ └── img_tools.py // 数据处理相关，包括数据归一化、扰动 │ │ ├── __init__.py // 构造 dataloader 相关代码
│ ├── __init__.py │ │ ├── lmdb_dataset.py // 读取lmdb数据集的 dataset
│ │ ├── simple_dataset.py // 读取文本格式存储数据集的 dataset
│ ├── losses // 损失函数
│ │ ├── __init__.py // 构造 loss 相关代码
│ │ ├── cls_loss.py // 方向分类器 loss
│ │ ├── det_basic_loss.py // 检测基础 loss
│ │ ├── det_db_loss.py // DB loss
│ │ ├── det_east_loss.py // EAST loss
│ │ ├── det_sast_loss.py // SAST loss
│ │ ├── rec_ctc_loss.py // CTC loss
│ │ ├── rec_att_loss.py // Attention loss
│ ├── metrics // 评估指标
│ │ ├── __init__.py // 构造 metric 相关代码
│ │ ├── cls_metric.py // 方向分类器 metric
│ │ ├── det_metric.py // 检测 metric
│ ├── eval_det_iou.py // 检测 iou 相关
│ │ ├── rec_metric.py // 识别 metric
│ ├── modeling // 组网相关 │ ├── modeling // 组网相关
│ │ ├── architectures // 模型架构，定义模型所需的各个模块 │ │ ├── architectures // 网络
│ │ │ ├── cls_model.py // 方向分类器 │ │ │ ├── __init__.py // 构造 model 相关代码
│ │ │ ├── det_model.py // 检测 │ │ │ ├── base_model.py // 组网代码
│ │ │ └── rec_model.py // 识别
│ │ ├── backbones // 骨干网络 │ │ ├── backbones // 骨干网络
│ │ │ ├── __init__.py // 构造 backbone 相关代码
│ │ │ ├── det_mobilenet_v3.py // 检测 mobilenet_v3 │ │ │ ├── det_mobilenet_v3.py // 检测 mobilenet_v3
│ │ │ ├── det_resnet_vd.py │ │ │ ├── det_resnet_vd.py // 检测 resnet
│ │ │ ├── det_resnet_vd_sast.py │ │ │ ├── det_resnet_vd_sast.py // 检测 SAST算法的resnet backbone
│ │ │ ├── rec_mobilenet_v3.py // 识别 mobilenet_v3 │ │ │ ├── rec_mobilenet_v3.py // 识别 mobilenet_v3
│ │ │ ├── rec_resnet_fpn.py │ │ │ └── rec_resnet_vd.py // 识别 resnet
│ │ │ └── rec_resnet_vd.py │ │ ├── necks // 颈函数
│ │ ├── common_functions.py // 公共函数 │ │ │ ├── __init__.py // 构造 neck 相关代码
│ │ │ ├── db_fpn.py // 标准 fpn 网络
│ │ │ ├── east_fpn.py // EAST 算法的 fpn 网络
│ │ │ ├── sast_fpn.py // SAST 算法的 fpn 网络
│ │ │ ├── rnn.py // 识别 序列编码
│ │ ├── heads // 头函数 │ │ ├── heads // 头函数
│ │ │ ├── cls_head.py // 分类头 │ │ │ ├── __init__.py // 构造 head 相关代码
│ │ │ ├── det_db_head.py // db 检测头 │ │ │ ├── cls_head.py // 方向分类器 分类头
│ │ │ ├── det_east_head.py // east 检测头 │ │ │ ├── det_db_head.py // DB 检测头
│ │ │ ├── det_sast_head.py // sast 检测头 │ │ │ ├── det_east_head.py // EAST 检测头
│ │ │ ├── rec_attention_head.py // 识别 attention │ │ │ ├── det_sast_head.py // SAST 检测头
│ │ │ ├── rec_ctc_head.py // 识别 ctc │ │ │ ├── rec_ctc_head.py // 识别 ctc
│ │ │ ├── rec_seq_encoder.py // 识别 序列编码 │ │ │ ├── rec_att_head.py // 识别 attention
│ │ │ ├── rec_srn_all_head.py // 识别 srn 相关 │ │ ├── transforms // 图像变换
│ │ │ └── self_attention // srn attention │ │ │ ├── __init__.py // 构造 transform 相关代码
│ │ │ └── model.py │ │ │ └── tps.py // TPS 变换
│ │ ├── losses // 损失函数 │ ├── optimizer // 优化器
│ │ │ ├── cls_loss.py // 方向分类器损失函数 │ │ ├── __init__.py // 构造 optimizer 相关代码
│ │ │ ├── det_basic_loss.py // 检测基础loss │ │ └── learning_rate.py // 学习率衰减
│ │ │ ├── det_db_loss.py // DB loss │ │ └── optimizer.py // 优化器
│ │ │ ├── det_east_loss.py // EAST loss │ │ └── regularizer.py // 网络正则化
│ │ │ ├── det_sast_loss.py // SAST loss
│ │ │ ├── rec_attention_loss.py // attention loss
│ │ │ ├── rec_ctc_loss.py // ctc loss
│ │ │ └── rec_srn_loss.py // srn loss
│ │ └── stns // 空间变换网络
│ │ └── tps.py // TPS 变换
│ ├── optimizer.py // 优化器
│ ├── postprocess // 后处理 │ ├── postprocess // 后处理
│ │ ├── cls_postprocess.py // 方向分类器 后处理
│ │ ├── db_postprocess.py // DB 后处理 │ │ ├── db_postprocess.py // DB 后处理
│ │ ├── east_postprocess.py // East 后处理 │ │ ├── east_postprocess.py // EAST 后处理
│ │ ├── lanms // lanms 相关 │ │ ├── locality_aware_nms.py // NMS
│ │ │ ... │ │ ├── rec_postprocess.py // 识别网络 后处理
│ │ ├── locality_aware_nms.py // nms │ │ └── sast_postprocess.py // SAST 后处理
│ │ └── sast_postprocess.py // sast 后处理
│ └── utils // 工具 │ └── utils // 工具
│ ├── character.py // 字符处理，包括对文本的编码和解码，计算预测准确率 │ ├── dict // 小语种字典
│ ├── check.py // 参数加载检查 │ ....
│ ├── ic15_dict.txt // 英文数字字典，区分大小写 │ ├── ic15_dict.txt // 英文数字字典，区分大小写
│ ├── ppocr_keys_v1.txt // 中文字典，用于训练中文模型 │ ├── ppocr_keys_v1.txt // 中文字典，用于训练中文模型
│ ├── logging.py // logger
│ ├── save_load.py // 模型保存和加载函数 │ ├── save_load.py // 模型保存和加载函数
│ ├── stats.py // 统计 │ ├── stats.py // 统计
│ └── utility.py // 工具函数，包含输入参数是否合法等相关检查工具 │ └── utility.py // 工具函数
├── README_en.md // 说明文档 ├── tools
├── README.md │ ├── eval.py // 评估函数
│ ├── export_model.py // 导出 inference 模型
│ ├── infer // 基于预测引擎预测
│ │ ├── predict_cls.py
│ │ ├── predict_det.py
│ │ ├── predict_rec.py
│ │ ├── predict_system.py
│ │ └── utility.py
│ ├── infer_cls.py // 基于训练引擎 预测分类
│ ├── infer_det.py // 基于训练引擎 预测检测
│ ├── infer_rec.py // 基于训练引擎 预测识别
│ ├── program.py // 整体流程
│ ├── test_hubserving.py
│ └── train.py // 启动训练
├── paddleocr.py
├── README_ch.md // 中文说明文档
├── README_en.md // 英文说明文档
├── README.md // 主页说明文档
├── requirments.txt // 安装依赖 ├── requirments.txt // 安装依赖
├── setup.py // whl包打包脚本 ├── setup.py // whl包打包脚本
└── tools // 启动工具 ├── train.sh // 启动训练脚本
├── eval.py // 评估函数 \ No newline at end of file
├── eval_utils // 评估工具
│ ├── eval_cls_utils.py // 分类相关
│ ├── eval_det_iou.py // 检测 iou 相关
│ ├── eval_det_utils.py // 检测相关
│ ├── eval_rec_utils.py // 识别相关
│ └── __init__.py
├── export_model.py // 导出 infer 模型
├── infer // 基于预测引擎预测
│ ├── predict_cls.py
│ ├── predict_det.py
│ ├── predict_rec.py
│ ├── predict_system.py
│ └── utility.py
├── infer_cls.py // 基于训练引擎 预测分类
├── infer_det.py // 基于训练引擎 预测检测
├── infer_rec.py // 基于训练引擎 预测识别
├── program.py // 整体流程
├── test_hubserving.py
└── train.py // 启动训练
```
doc/doc_en/tree_en.md
浏览文件 @ 91f5ab5c
...@@ -2,40 +2,37 @@ ...@@ -2,40 +2,37 @@
The overall directory structure of PaddleOCR is introduced as follows: The overall directory structure of PaddleOCR is introduced as follows:
``` ```
PaddleOCR PaddleOCR
├── configs // configuration file, you can select model structure and modify hyperparameters through yml file ├── configs // Configuration file, you can config the model structure and modify the hyperparameters through the yml file
│ ├── cls // Related configuration files of direction classifier │ ├── cls // Angle classifier config files
│ │ ├── cls_mv3.yml // training configuration related, including backbone network, head, loss, optimizer │ │ ├── cls_mv3.yml // Training config, including backbone network, head, loss, optimizer and data
│ │ └── cls_reader.yml // Data reading related, data reading method, data storage path │ ├── det // Text detection config files
│ ├── det // Detection related configuration files │ │ ├── det_mv3_db.yml // Training config
│ │ ├── det_db_icdar15_reader.yml // data read
│ │ ├── det_mv3_db.yml // training configuration
│ │ ... │ │ ...
│ └── rec // Identify related configuration files │ └── rec // Text recognition config files
│ ├── rec_benchmark_reader.yml // LMDB format data reading related │ ├── rec_mv3_none_bilstm_ctc.yml // CRNN config
│ ├── rec_chinese_common_train.yml // General Chinese training configuration
│ ├── rec_icdar15_reader.yml // simple data reading related, including data reading function, data path, label file
│ ... │ ...
├── deploy // deployment related ├── deploy // Depoly
│ ├── android_demo // android_demo │ ├── android_demo // Android demo
│ │ ... │ │ ...
│ ├── cpp_infer // C++ infer │ ├── cpp_infer // C++ infer
│ │ ├── CMakeLists.txt // Cmake file │ │ ├── CMakeLists.txt // Cmake file
│ │ ├── docs // documentation │ │ ├── docs // Docs
│ │ │ └── windows_vs2019_build.md │ │ │ └── windows_vs2019_build.md
│ │ ├── include │ │ ├── include // Head Files
│ │ │ ├── clipper.h // clipper library │ │ │ ├── clipper.h // clipper
│ │ │ ├── config.h // infer configuration │ │ │ ├── config.h // Inference config
│ │ │ ├── ocr_cls.h // direction classifier │ │ │ ├── ocr_cls.h // Angle class
│ │ │ ├── ocr_det.h // text detection │ │ │ ├── ocr_det.h // Text detection
│ │ │ ├── ocr_rec.h // text recognition │ │ │ ├── ocr_rec.h // Text recognition
│ │ │ ├── postprocess_op.h // postprocess after detection │ │ │ ├── postprocess_op.h // Post-processing
│ │ │ ├── preprocess_op.h // preprocess detection │ │ │ ├── preprocess_op.h // Pre-processing
│ │ │ └── utility.h // tools │ │ │ └── utility.h // tools
│ │ ├── readme.md // documentation │ │ ├── readme.md // Documentation
│ │ ├── ... │ │ ├── ...
│ │ ├── src // source file │ │ ├── src // Source code files
│ │ │ ├── clipper.cpp │ │ │ ├── clipper.cpp
│ │ │ ├── config.cpp │ │ │ ├── config.cpp
│ │ │ ├── main.cpp │ │ │ ├── main.cpp
...@@ -45,10 +42,10 @@ PaddleOCR ...@@ -45,10 +42,10 @@ PaddleOCR
│ │ │ ├── postprocess_op.cpp │ │ │ ├── postprocess_op.cpp
│ │ │ ├── preprocess_op.cpp │ │ │ ├── preprocess_op.cpp
│ │ │ └── utility.cpp │ │ │ └── utility.cpp
│ │ └── tools // compile and execute script │ │ └── tools // Compile and execute script
│ │ ├── build.sh // compile script │ │ ├── build.sh // Compile script
│ │ ├── config.txt // configuration file │ │ ├── config.txt // Config file
│ │ └── run.sh // Test startup script │ │ └── run.sh // Execute script
│ ├── docker │ ├── docker
│ │ └── hubserving │ │ └── hubserving
│ │ ├── cpu │ │ ├── cpu
...@@ -59,150 +56,164 @@ PaddleOCR ...@@ -59,150 +56,164 @@ PaddleOCR
│ │ ├── README.md │ │ ├── README.md
│ │ └── sample_request.txt │ │ └── sample_request.txt
│ ├── hubserving // hubserving │ ├── hubserving // hubserving
│ │ ├── ocr_det // text detection │ │ ├── ocr_cls // Angle class
│ │ │ ├── config.json // serving configuration │ │ │ ├── config.json // Serving config
│ │ │ ├── __init__.py
│ │ │ ├── module.py // Model
│ │ │ └── params.py // Parameters
│ │ ├── ocr_det // Text detection
│ │ │ ├── config.json // serving config
│ │ │ ├── __init__.py │ │ │ ├── __init__.py
│ │ │ ├── module.py // prediction model │ │ │ ├── module.py // Model
│ │ │ └── params.py // prediction parameters │ │ │ └── params.py // Parameters
│ │ ├── ocr_rec // text recognition │ │ ├── ocr_rec // Text recognition
│ │ │ ├── config.json │ │ │ ├── config.json
│ │ │ ├── __init__.py │ │ │ ├── __init__.py
│ │ │ ├── module.py │ │ │ ├── module.py
│ │ │ └── params.py │ │ │ └── params.py
│ │ └── ocr_system // system forecast │ │ └── ocr_system // Inference System
│ │ ├── config.json │ │ ├── config.json
│ │ ├── __init__.py │ │ ├── __init__.py
│ │ ├── module.py │ │ ├── module.py
│ │ └── params.py │ │ └── params.py
│ ├── imgs // prediction picture │ ├── imgs // Inference images
│ │ ├── cpp_infer_pred_12.png │ │ ├── cpp_infer_pred_12.png
│ │ └── demo.png │ │ └── demo.png
│ ├── ios_demo // ios demo │ ├── ios_demo // IOS demo
│ │ ... │ │ ...
│ ├── lite // lite deployment │ ├── lite // Lite depoly
│ │ ├── cls_process.cc // direction classifier data processing │ │ ├── cls_process.cc // Pre-process for angle class
│ │ ├── cls_process.h │ │ ├── cls_process.h
│ │ ├── config.txt // check configuration parameters │ │ ├── config.txt // Config file
│ │ ├── crnn_process.cc // crnn data processing │ │ ├── crnn_process.cc // Pre-process for CRNN
│ │ ├── crnn_process.h │ │ ├── crnn_process.h
│ │ ├── db_post_process.cc // db data processing │ │ ├── db_post_process.cc // Pre-process for DB
│ │ ├── db_post_process.h │ │ ├── db_post_process.h
│ │ ├── Makefile // compile file │ │ ├── Makefile // Compile file
│ │ ├── ocr_db_crnn.cc // series prediction │ │ ├── ocr_db_crnn.cc // Inference system
│ │ ├── prepare.sh // data preparation │ │ ├── prepare.sh // Prepare bash script
│ │ ├── readme.md // documentation │ │ ├── readme.md // Documentation
│ │ ... │ │ ...
│ ├── pdserving // pdserving deployment │ ├── pdserving // Pdserving depoly
│ │ ├── det_local_server.py // fast detection version, easy deployment and fast prediction │ │ ├── det_local_server.py // Text detection fast version, easy to deploy and fast to predict
│ │ ├── det_web_server.py // Full version of detection, high stability and distributed deployment │ │ ├── det_web_server.py // Text detection full version, high stability distributed deployment
│ │ ├── ocr_local_server.py // detection + identification quick version │ │ ├── ocr_local_server.py // Text detection + recognition fast version
│ │ ├── ocr_web_client.py // client │ │ ├── ocr_web_client.py // client
│ │ ├── ocr_web_server.py // detection + identification full version │ │ ├── ocr_web_server.py // Text detection + recognition full version
│ │ ├── readme.md // documentation │ │ ├── readme.md // Documentation
│ │ ├── rec_local_server.py // recognize quick version │ │ ├── rec_local_server.py // Text recognition fast version
│ │ └── rec_web_server.py // Identify the full version │ │ └── rec_web_server.py // Text recognition full version
│ └── slim │ └── slim
│ └── quantization // quantization related │ └── quantization // Quantization
│ ├── export_model.py // export model │ ├── export_model.py // Export model
│ ├── quant.py // quantization │ ├── quant.py // Quantization script
│ └── README.md // Documentation │ └── README.md // Documentation
├── doc // Documentation tutorial ├── doc // Documentation and Tutorials
│ ... │ ...
├── paddleocr.py ├── ppocr // Core code
├── ppocr // network core code │ ├── data // Data processing
│ ├── data // data processing │ │ ├── imaug // Image and label processing code
│ │ ├── cls // direction classifier │ │ │ ├── text_image_aug // Tia data augment for text recognition
│ │ │ ├── dataset_traversal.py // Data transmission, define data reader, read data and form batch │ │ │ │ ├── __init__.py
│ │ │ └── randaugment.py // Random data augmentation operation │ │ │ │ ├── augment.py // Tia_distort,tia_stretch and tia_perspective
│ │ ├── det // detection │ │ │ │ ├── warp_mls.py
│ │ │ ├── data_augment.py // data augmentation operation │ │ │ ├── __init__.py
│ │ │ ├── dataset_traversal.py // Data transmission, define data reader, read data and form batch │ │ │ ├── east_process.py // Data processing steps of EAST algorithm
│ │ │ ├── db_process.py // db data processing │ │ │ ├── iaa_augment.py // Data augmentation operations
│ │ │ ├── east_process.py // east data processing │ │ │ ├── label_ops.py // label encode operations
│ │ │ ├── make_border_map.py // Generate boundary map │ │ │ ├── make_border_map.py // Generate boundary map
│ │ │ ├── make_shrink_map.py // Generate shrink map │ │ │ ├── make_shrink_map.py // Generate shrink graph
│ │ │ ├── random_crop_data.py // random crop │ │ │ ├── operators.py // Basic image operations, such as reading and normalization
│ │ │ └── sast_process.py // sast data processing │ │ │ ├── randaugment.py // Random data augmentation operation
│ │ ├── reader_main.py // main function of data reader │ │ │ ├── random_crop_data.py // Random crop
│ │ └── rec // recognation │ │ │ ├── rec_img_aug.py // Data augmentation for text recognition
│ │ ├── dataset_traversal.py // Data transmission, define data reader, including LMDB_Reader and Simple_Reader │ │ │ └── sast_process.py // Data processing steps of SAST algorithm
│ │ └── img_tools.py // Data processing related, including data normalization and disturbance │ │ ├── __init__.py // Construct dataloader code
│ ├── __init__.py │ │ ├── lmdb_dataset.py // Read lmdb dataset
│ ├── modeling // networking related │ │ ├── simple_dataset.py // Read the dataset stored in text format
│ │ ├── architectures // Model architecture, which defines the various modules required by the model │ ├── losses // Loss function
│ │ │ ├── cls_model.py // direction classifier │ │ ├── __init__.py // Construct loss code
│ │ │ ├── det_model.py // detection │ │ ├── cls_loss.py // Angle class loss
│ │ │ └── rec_model.py // recognition │ │ ├── det_basic_loss.py // Text detection basic loss
│ │ ├── backbones // backbone network │ │ ├── det_db_loss.py // DB loss
│ │ │ ├── det_mobilenet_v3.py // detect mobilenet_v3 │ │ ├── det_east_loss.py // EAST loss
│ │ │ ├── det_resnet_vd.py │ │ ├── det_sast_loss.py // SAST loss
│ │ │ ├── det_resnet_vd_sast.py │ │ ├── rec_ctc_loss.py // CTC loss
│ │ │ ├── rec_mobilenet_v3.py // recognize mobilenet_v3 │ │ ├── rec_att_loss.py // Attention loss
│ │ │ ├── rec_resnet_fpn.py │ ├── metrics // Metrics
│ │ │ └── rec_resnet_vd.py │ │ ├── __init__.py // Construct metric code
│ │ ├── common_functions.py // common functions │ │ ├── cls_metric.py // Angle class metric
│ │ ├── heads │ │ ├── det_metric.py // Text detection metric
│ │ │ ├── cls_head.py // class header │ ├── eval_det_iou.py // Text detection iou code
│ │ │ ├── det_db_head.py // db detection head │ │ ├── rec_metric.py // Text recognition metric
│ │ │ ├── det_east_head.py // east detection head │ ├── modeling // Network
│ │ │ ├── det_sast_head.py // sast detection head │ │ ├── architectures // Architecture
│ │ │ ├── rec_attention_head.py // recognition attention │ │ │ ├── __init__.py // Construct model code
│ │ │ ├── rec_ctc_head.py // recognition ctc │ │ │ ├── base_model.py // Base model
│ │ │ ├── rec_seq_encoder.py // recognition sequence code │ │ ├── backbones // backbones
│ │ │ ├── rec_srn_all_head.py // srn related │ │ │ ├── __init__.py // Construct backbone code
│ │ │ └── self_attention // srn attention │ │ │ ├── det_mobilenet_v3.py // Text detection mobilenet_v3
│ │ │ └── model.py │ │ │ ├── det_resnet_vd.py // Text detection resnet
│ │ ├── losses // loss function │ │ │ ├── det_resnet_vd_sast.py // Text detection resnet backbone of the SAST algorithm
│ │ │ ├── cls_loss.py // Directional classifier loss function │ │ │ ├── rec_mobilenet_v3.py // Text recognition mobilenet_v3
│ │ │ ├── det_basic_loss.py // detect basic loss │ │ │ └── rec_resnet_vd.py // Text recognition resnet
│ │ │ ├── det_db_loss.py // DB loss │ │ ├── necks // Necks
│ │ │ ├── det_east_loss.py // EAST loss │ │ │ ├── __init__.py // Construct neck code
│ │ │ ├── det_sast_loss.py // SAST loss │ │ │ ├── db_fpn.py // Standard fpn
│ │ │ ├── rec_attention_loss.py // attention loss │ │ │ ├── east_fpn.py // EAST algorithm fpn network
│ │ │ ├── rec_ctc_loss.py // ctc loss │ │ │ ├── sast_fpn.py // SAST algorithm fpn network
│ │ │ └── rec_srn_loss.py // srn loss │ │ │ ├── rnn.py // Character recognition sequence encoding
│ │ └── stns // Spatial transformation network │ │ ├── heads // Heads
│ │ └── tps.py // TPS conversion │ │ │ ├── __init__.py // Construct head code
│ ├── optimizer.py // optimizer │ │ │ ├── cls_head.py // Angle class head
│ ├── postprocess // post-processing │ │ │ ├── det_db_head.py // DB head
│ │ ├── db_postprocess.py // DB postprocess │ │ │ ├── det_east_head.py // EAST head
│ │ ├── east_postprocess.py // East postprocess │ │ │ ├── det_sast_head.py // SAST head
│ │ ├── lanms // lanms related │ │ │ ├── rec_ctc_head.py // CTC head
│ │ │ ... │ │ │ ├── rec_att_head.py // Attention head
│ │ ├── locality_aware_nms.py // nms │ │ ├── transforms // Transforms
│ │ └── sast_postprocess.py // sast post-processing │ │ │ ├── __init__.py // Construct transform code
│ └── utils // tools │ │ │ └── tps.py // TPS transform
│ ├── character.py // Character processing, including text encoding and decoding, and calculation of prediction accuracy │ ├── optimizer // Optimizer
│ ├── check.py // parameter loading check │ │ ├── __init__.py // Construct optimizer code
│ │ └── learning_rate.py // Learning rate decay
│ │ └── optimizer.py // Optimizer
│ │ └── regularizer.py // Network regularization
│ ├── postprocess // Post-processing
│ │ ├── cls_postprocess.py // Angle class post-processing
│ │ ├── db_postprocess.py // DB post-processing
│ │ ├── east_postprocess.py // EAST post-processing
│ │ ├── locality_aware_nms.py // NMS
│ │ ├── rec_postprocess.py // Text recognition post-processing
│ │ └── sast_postprocess.py // SAST post-processing
│ └── utils // utils
│ ├── dict // Minor language dictionary
│ ....
│ ├── ic15_dict.txt // English number dictionary, case sensitive │ ├── ic15_dict.txt // English number dictionary, case sensitive
│ ├── ppocr_keys_v1.txt // Chinese dictionary, used to train Chinese models │ ├── ppocr_keys_v1.txt // Chinese dictionary for training Chinese models
│ ├── save_load.py // model save and load function │ ├── logging.py // logger
│ ├── stats.py // Statistics │ ├── save_load.py // Model saving and loading functions
│ └── utility.py // Tool functions, including related check tools such as whether the input parameters are legal │ ├── stats.py // Training status statistics
├── README_en.md // documentation │ └── utility.py // Utility function
├── README.md ├── tools
├── requirments.txt // installation dependencies │ ├── eval.py // Evaluation function
├── setup.py // whl package packaging script │ ├── export_model.py // Export inference model
└── tools // start tool │ ├── infer // Inference based on Inference engine
├── eval.py // evaluation function │ │ ├── predict_cls.py
├── eval_utils // evaluation tools │ │ ├── predict_det.py
│ ├── eval_cls_utils.py // category related │ │ ├── predict_rec.py
│ ├── eval_det_iou.py // detect iou related │ │ ├── predict_system.py
│ ├── eval_det_utils.py // detection related │ │ └── utility.py
│ ├── eval_rec_utils.py // recognition related │ ├── infer_cls.py // Angle classification inference based on training engine
│ └── __init__.py │ ├── infer_det.py // Text detection inference based on training engine
├── export_model.py // export infer model │ ├── infer_rec.py // Text recognition inference based on training engine
├── infer // Forecast based on prediction engine │ ├── program.py // Inference system
│ ├── predict_cls.py │ ├── test_hubserving.py
│ ├── predict_det.py │ └── train.py // Start training script
│ ├── predict_rec.py ├── paddleocr.py
│ ├── predict_system.py ├── README_ch.md // Chinese documentation
│ └── utility.py ├── README_en.md // English documentation
├── infer_cls.py // Predict classification based on training engine ├── README.md // Home page documentation
├── infer_det.py // Predictive detection based on training engine ├── requirments.txt // Requirments
├── infer_rec.py // Predictive recognition based on training engine ├── setup.py // Whl package packaging script
├── program.py // overall process ├── train.sh // Start training bash script
├── test_hubserving.py \ No newline at end of file
└── train.py // start training
```
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或