guanshuicheng / invoice

Collaboration with wangxupeng(https://github.com/wangxupeng)
MIT License
1.82k stars 454 forks source link

请问训练的脚本在哪里呢 #29

Open sideme opened 4 years ago

sideme commented 4 years ago

刚刚接触这块,不知道你是如何训练的,以及训练的代码在哪儿?

guanshuicheng commented 4 years ago

刚刚接触这块,不知道你是如何训练的,以及训练的代码在哪儿?

检测部分训练脚本:https://github.com/chineseocr/chineseocr/blob/master/train/text/text-train.ipynb 识别部分训练脚本:https://github.com/chineseocr/chineseocr/blob/master/train/ocr/train-ocr.ipynb

sideme commented 4 years ago

刚刚接触这块,不知道你是如何训练的,以及训练的代码在哪儿?

检测部分训练脚本:https://github.com/chineseocr/chineseocr/blob/master/train/text/text-train.ipynb 识别部分训练脚本:https://github.com/chineseocr/chineseocr/blob/master/train/ocr/train-ocr.ipynb

如果我想增加识别区域,是不是我只要用labelImg标注多个区域,然后再按照你上面写的这两个脚本训练一下就可以了呢?我现在用labelImg标注了以后,尝试用yolov3去训练,不知道这样可不可行

guanshuicheng commented 3 years ago

刚刚接触这块,不知道你是如何训练的,以及训练的代码在哪儿?

检测部分训练脚本:https://github.com/chineseocr/chineseocr/blob/master/train/text/text-train.ipynb 识别部分训练脚本:https://github.com/chineseocr/chineseocr/blob/master/train/ocr/train-ocr.ipynb

如果我想增加识别区域,是不是我只要用labelImg标注多个区域,然后再按照你上面写的这两个脚本训练一下就可以了呢?我现在用labelImg标注了以后,尝试用yolov3去训练,不知道这样可不可行

box检测模型标注软件参考https://github.com/cgvict/roLabelImg.git,可标注带角度的box,标注完可直接训练yolov3的检测模型