Functions — github.com/bai-shang/crnn_seq2seq_ocr_pytorch

CodeHub MCPcopy Create free account

hub / github.com/bai-shang/crnn_seq2seq_ocr_pytorch / functions

Functions36 in github.com/bai-shang/crnn_seq2seq_ocr_pytorch

⨍ Functions 36 ◇ Types & classes 11

↓ 4 callersMethod__init__

(self, channel_size)

crnn/seq2seq.py:12

↓ 3 callersMethodinitHidden

(self, batch_size)

crnn/seq2seq.py:115

↓ 2 callersMethoddecode

Decode encoded texts back into strs. Args: torch.IntTensor [length_0 + length_1 + ... length_{n - 1}]: encoded texts.

src/utils.py:57

↓ 2 callersMethodencode

Args: text (str or list of str): texts to convert. Returns: torch.IntTensor targets:max_length × batch_size

src/utils.py:34

↓ 2 callersMethodreset

src/utils.py:91

↓ 2 callersMethodval

src/utils.py:95

↓ 1 callersMethodadd

src/utils.py:80

↓ 1 callersFunctionevaluate

(image, text, encoder, decoder, data_loader, max_eval_iter=100)

↓ 1 callersFunctionmain

↓ 1 callersFunctionmain

inference.py:59

↓ 1 callersFunctionseq2seq_decode

(encoder_out, decoder, decoder_input, decoder_hidden, max_length)

inference.py:38

↓ 1 callersFunctiontrain

(image, text, encoder, decoder, criterion, train_loader, teach_forcing_prob=1)

src/dataset.py:55

src/dataset.py:106

Method__getitem__

src/dataset.py:24

(self, input_size, hidden_size, output_size)

crnn/seq2seq.py:31

(self, hidden_size, output_size, dropout_p=0.1, max_length=71)

crnn/seq2seq.py:49

(self, channel_size, hidden_size)

crnn/seq2seq.py:85

(self, hidden_size, output_size, dropout_p=0.1, max_length=71)

crnn/seq2seq.py:107

(self, alphabet)

src/utils.py:25

src/utils.py:77

(self, text_line_file=None, transform=None, target_transform=None)

src/dataset.py:12

(self, img_width, img_height)

src/dataset.py:50

(self, data_source, batch_size)

src/dataset.py:75

(self, img_height=32, img_width=100)

src/dataset.py:101

src/dataset.py:79

src/dataset.py:21

src/dataset.py:95

crnn/seq2seq.py:23

crnn/seq2seq.py:37

(self, input, hidden, encoder_outputs)

crnn/seq2seq.py:64

crnn/seq2seq.py:92

(self, input, hidden, encoder_outputs)

crnn/seq2seq.py:112

MethodinitHidden

crnn/seq2seq.py:80

Functionload_data

src/utils.py:101

Functionweights_init

src/utils.py:106