-
大小: 51.15MB文件類型: .zip金幣: 2下載: 0 次發布日期: 2024-01-29
- 語言: 其他
- 標簽: OCR??yolov3+ocr??cv??
資源簡介
自然場景OCR(YOLOv3+CTPN+CRNN),附帶整個項目代碼,詳細代碼注釋
代碼片段和文件信息
#?-*-?coding:?utf-8?-*-
“““
@author:?lywen
“““
import?os
import?cv2
import?json
import?time
import?uuid
import?base64
import?web
from?PIL?import?Image
web.config.debug??=?True
import?model
render?=?web.template.render(‘templates‘?base=‘base‘)
from?config?import?DETECTANGLE
from?apphelper.image?import?union_rboxadjust_box_to_origin
from?application?import?trainTicketidcard?
billList?=?[‘通用OCR‘‘火車票‘‘身份證‘]
class?OCR:
????“““通用OCR識別“““
????def?GET(self):
????????post?=?{}
????????post[‘postName‘]?=?‘ocr‘##請求地址
????????post[‘height‘]?=?1000
????????post[‘H‘]?=?1000
????????post[‘width‘]?=?600
????????post[‘W‘]?=?600
????????post[‘uuid‘]?=?uuid.uuid1().__str__()
????????post[‘billList‘]?=?billList
????????return?render.ocr(post)
????def?POST(self):
????????data?=?web.data()
????????data?=?json.loads(data)
????????billModel?=?data.get(‘billModel‘‘‘)
????????textAngle?=?data.get(‘textAngle‘False)##文字檢測
????????textLine?=?data.get(‘textLine‘False)##只進行單行識別
????????
????????imgString?=?data[‘imgString‘].encode().split(b‘;base64‘)[-1]
????????imgString?=?base64.b64decode(imgString)
????????jobid?=?uuid.uuid1().__str__()
????????path?=?‘test/{}.jpg‘.format(jobid)
????????with?open(path‘wb‘)?as?f:
????????????f.write(imgString)
????????img?=?cv2.imread(path)##GBR
????????HW?=?img.shape[:2]
????????timetake?=?time.time()
????????if?textLine:
????????????##單行識別
????????????partImg?=?Image.fromarray(img)
????????????text?=?model.crnnOcr(partImg.convert(‘L‘))
????????????res?=[?{‘text‘:text‘name‘:‘0‘‘box‘:[00W0WH0H]}?]
????????else:
????????????detectAngle?=?textAngle
????????????_resultangle=?model.model(img
????????????????????????????????????????detectAngle=detectAngle##是否進行文字方向檢測,通過web傳參控制
????????????????????????????????????????config=dict(MAX_HORIZONTAL_GAP=50##字符之間的最大間隔,用于文本行的合并
????????????????????????????????????????MIN_V_OVERLAPS=0.6
????????????????????????????????????????MIN_SIZE_SIM=0.6
????????????????????????????????????????TEXT_PROPOSALS_MIN_SCORE=0.1
????????????????????????????????????????TEXT_PROPOSALS_NMS_THRESH=0.3
????????????????????????????????????????TEXT_LINE_NMS_THRESH?=?0.7##文本行之間測iou值
????????????????????????????????????????????????)
????????????????????????????????????????leftAdjust=True##對檢測的文本行進行向左延伸
????????????????????????????????????????rightAdjust=True##對檢測的文本行進行向右延伸
????????????????????????????????????????alph=0.01##對檢測的文本行進行向右、左延伸的倍數
???????????????????????????????????????)
????????????if?billModel==‘‘?or?billModel==‘通用OCR‘?:
????????????????result?=?union_rbox(result0.2)
????????????????res?=?[{‘text‘:x[‘text‘]
????????????????????????‘name‘:str(i)
????????????????????????‘box‘:{‘cx‘:x[‘cx‘]
???????????????????????????????‘cy‘:x[‘cy‘]
???????????????????????????????‘w‘:x[‘w‘]
???????????????????????????????‘h‘:x[‘h‘]
???????????????????????????????‘angle‘:x[‘degree‘]
??????????????????????????????}
???????????????????????}?for?ix?in?enumerate(result)]
????????????????res?=?a
?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
?????目錄???????????0??2019-05-12?15:40??chineseocr\
?????目錄???????????0??2019-04-15?18:52??chineseocr\.git\
?????文件??????????23??2019-04-15?18:52??chineseocr\.git\HEAD
?????目錄???????????0??2019-04-15?18:52??chineseocr\.git\branches\
?????文件?????????269??2019-04-15?18:52??chineseocr\.git\config
?????文件??????????73??2019-04-15?18:52??chineseocr\.git\desc
?????目錄???????????0??2019-04-15?18:52??chineseocr\.git\hooks\
?????文件?????????478??2019-04-15?18:52??chineseocr\.git\hooks\applypatch-msg.sample
?????文件?????????896??2019-04-15?18:52??chineseocr\.git\hooks\commit-msg.sample
?????文件?????????189??2019-04-15?18:52??chineseocr\.git\hooks\post-update.sample
?????文件?????????424??2019-04-15?18:52??chineseocr\.git\hooks\pre-applypatch.sample
?????文件????????1642??2019-04-15?18:52??chineseocr\.git\hooks\pre-commit.sample
?????文件????????1348??2019-04-15?18:52??chineseocr\.git\hooks\pre-push.sample
?????文件????????4898??2019-04-15?18:52??chineseocr\.git\hooks\pre-reba
?????文件????????1239??2019-04-15?18:52??chineseocr\.git\hooks\prepare-commit-msg.sample
?????文件????????3610??2019-04-15?18:52??chineseocr\.git\hooks\update.sample
?????文件???????27391??2019-04-15?18:52??chineseocr\.git\index
?????目錄???????????0??2019-04-15?18:52??chineseocr\.git\info\
?????文件?????????240??2019-04-15?18:52??chineseocr\.git\info\exclude
?????目錄???????????0??2019-04-15?18:52??chineseocr\.git\logs\
?????文件?????????205??2019-04-15?18:52??chineseocr\.git\logs\HEAD
?????目錄???????????0??2019-04-15?18:52??chineseocr\.git\logs\refs\
?????目錄???????????0??2019-04-15?18:52??chineseocr\.git\logs\refs\heads\
?????文件?????????205??2019-04-15?18:52??chineseocr\.git\logs\refs\heads\master
?????目錄???????????0??2019-04-15?18:52??chineseocr\.git\logs\refs\remotes\
?????目錄???????????0??2019-04-15?18:52??chineseocr\.git\logs\refs\remotes\origin\
?????文件?????????205??2019-04-15?18:52??chineseocr\.git\logs\refs\remotes\origin\HEAD
?????目錄???????????0??2019-04-15?18:52??chineseocr\.git\ob
?????目錄???????????0??2019-04-15?18:52??chineseocr\.git\ob
?????目錄???????????0??2019-04-15?18:52??chineseocr\.git\ob
?????文件???????25124??2019-04-15?18:52??chineseocr\.git\ob
............此處省略390個文件信息
評論
共有 條評論