PaddleOCR在训练模型时,按照官方的指导教程,如下图所示(以ppocr_v2.0中的rec_chinese_common_train_v2.0.yml配置文件为例),一般是在终端输入一下命令行的方式配置运行进行训练。
python tools/train.py -c configs/rec/ch_ppocr_v2.0/rec_chinese_common_train_v2.0.yml
那么如果想通过tools\train.py在右键运行或进入debug模式,就需要修改如下代码:
①首先找到tools\program.py文件,打开该文件后,第45行代码处:
②对该行代码进行修改,具体为在其中添加一default参数,该参数为配置文件的绝对文件位置。替换如下代码(文件位置需要与你自己的匹配)。
self.add_argument("-c", "--config", default='G:/Code/python/paddleOCR/configs/rec/ch_ppocr_v2.0/rec_chinese_common_train_v2.0.yml', help="configuration file to use")
③修改相应的配置文件,主要是预训练模型、字典、训练数据集和验证数据集的路径(均需改为绝对路径),如下图所示:
完成后就可以自己debug进行调试了。