• 大小: 16KB
    文件类型: .py
    金币: 1
    下载: 0 次
    发布日期: 2021-01-03
  • 语言: Python
  • 标签: crnn  tensorflow  ctc  

资源简介

cnn +rnn +attention 以及CTC-loss融合的文字识别代码,基于tensorflow实现,要的拿去不客气,样本使用自我合成的数据,可自己添加,有些人傻的,上来就说不能直接运行,大爷的说了要添加训练样本,傻吗。。。

资源截图

代码片段和文件信息

#!/usr/bin/env python2
# -*- coding: utf-8 -*-
“““
crnn letter num  0.0001

“““

# from gen_hanzi import *
from tensorflow.contrib import layers
from tensorflow.python.layers.core import Dense
import numpy as np
import time
import os
import tensorflow as tf
import pdb
import cv2
from data_get_ctc_att import *

os.environ[“CUDA_VISIBLE_DEVICES“] = ‘1‘

obj = data_get(‘./img‘)

att_int_to_vocab att_vocab_to_int ctc_v2i = obj.extract_character_vocab()
# special_words = [‘‘ ‘‘ ‘‘  ‘‘]
START_TOKEN = 2
END_TOKEN = 3
UNK_TOKEN = 1
VOCAB_ATT = att_vocab_to_int
VOCAB_ATT_SIZE = len(VOCAB_ATT)  # 3840 + 4
VOCAB_CTC = ctc_v2i
VOCAB_CTC_SIZE = len(VOCAB_CTC)  # when used need + 1     # 3840

TRANSLATION = att_int_to_vocab

(IMHIGHT IMWIDTH) = (32 256)
ATT_embed_DIM = 512
MAXI

评论

共有 条评论