11 2021 档案
摘要:1. 打开 txt 文件,构建词典 with open('../data/OCR-data/test.txt') as f: data = f.readlines() alphabet=[x.rstrip() for x in data] alphabet=''.join(alphabet) 第一步
阅读全文
摘要:1. 整理数据集 将 [image] 和 [label]分开,其中image是gray格式图片矩阵,label是str class listDataset(Dataset): def __init__(self,list_file=None,transform=None,target_transfo
阅读全文
摘要:1. 在enocder_output上计算注意力权重 class AttentiondecoderV2(nn.Module): """ 采用seq2seq """ def __init__(self,hidden_size,output_size,dropout_p=0.1): super(Atte
阅读全文
摘要:1. 双向LSTM net:[in,hidden,out] [256,256,10] input:[w,h,in] [16,16,256] output:[w,h,out] [16,16,10] class BidirectionalLSTM(nn.Module): def __init__(sel
阅读全文