├── .gitignore ├── README.md ├── data ├── char_std_5990.txt └── demo.png ├── demo.py ├── expr └── attentioncnn │ └── encoder_600.pth ├── models ├── __init__.py ├── crnn.py └── crnn_lang.py ├── requirements.txt ├── src ├── __init__.py ├── class_attention.py ├── dataset.py └── utils.py ├── test_img ├── 20436312_1683447152.jpg ├── 20437109_1639581473.jpg ├── 20437421_2143654630.jpg ├── 20437531_1514396900.jpg ├── 20438953_2386326516.jpg ├── 20438984_3963047307.jpg ├── 20439171_260546633.jpg ├── 20439281_953270478.jpg ├── 20439562_2199433254.jpg ├── 20439906_2889507409.jpg ├── 20440000_4153137105.jpg ├── 20441140_1981970116.jpg ├── 20441328_2992333319.jpg ├── 20441531_4212871437.jpg ├── 20441750_1134599790.jpg ├── 20441765_2389222735.jpg ├── 20441859_4060744030.jpg ├── 20441968_1253178421.jpg ├── 20442421_1459247642.jpg ├── 20442468_3479649590.jpg ├── 20442515_3243295335.jpg ├── 20442640_2409912250.jpg ├── 20442750_205861976.jpg ├── 20442984_333319517.jpg └── md_img │ ├── attentionV2.png │ ├── attentionocr.png │ └── attention结果.png └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/README.md -------------------------------------------------------------------------------- /data/char_std_5990.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/data/char_std_5990.txt -------------------------------------------------------------------------------- /data/demo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/data/demo.png -------------------------------------------------------------------------------- /demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/demo.py -------------------------------------------------------------------------------- /expr/attentioncnn/encoder_600.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/expr/attentioncnn/encoder_600.pth -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/crnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/models/crnn.py -------------------------------------------------------------------------------- /models/crnn_lang.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/models/crnn_lang.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/class_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/src/class_attention.py -------------------------------------------------------------------------------- /src/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/src/dataset.py -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/src/utils.py -------------------------------------------------------------------------------- /test_img/20436312_1683447152.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/test_img/20436312_1683447152.jpg -------------------------------------------------------------------------------- /test_img/20437109_1639581473.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/test_img/20437109_1639581473.jpg -------------------------------------------------------------------------------- /test_img/20437421_2143654630.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/test_img/20437421_2143654630.jpg -------------------------------------------------------------------------------- /test_img/20437531_1514396900.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/test_img/20437531_1514396900.jpg -------------------------------------------------------------------------------- /test_img/20438953_2386326516.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/test_img/20438953_2386326516.jpg -------------------------------------------------------------------------------- /test_img/20438984_3963047307.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/test_img/20438984_3963047307.jpg -------------------------------------------------------------------------------- /test_img/20439171_260546633.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/test_img/20439171_260546633.jpg -------------------------------------------------------------------------------- /test_img/20439281_953270478.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/test_img/20439281_953270478.jpg -------------------------------------------------------------------------------- /test_img/20439562_2199433254.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/test_img/20439562_2199433254.jpg -------------------------------------------------------------------------------- /test_img/20439906_2889507409.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/test_img/20439906_2889507409.jpg -------------------------------------------------------------------------------- /test_img/20440000_4153137105.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/test_img/20440000_4153137105.jpg -------------------------------------------------------------------------------- /test_img/20441140_1981970116.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/test_img/20441140_1981970116.jpg -------------------------------------------------------------------------------- /test_img/20441328_2992333319.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/test_img/20441328_2992333319.jpg -------------------------------------------------------------------------------- /test_img/20441531_4212871437.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/test_img/20441531_4212871437.jpg -------------------------------------------------------------------------------- /test_img/20441750_1134599790.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/test_img/20441750_1134599790.jpg -------------------------------------------------------------------------------- /test_img/20441765_2389222735.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/test_img/20441765_2389222735.jpg -------------------------------------------------------------------------------- /test_img/20441859_4060744030.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/test_img/20441859_4060744030.jpg -------------------------------------------------------------------------------- /test_img/20441968_1253178421.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/test_img/20441968_1253178421.jpg -------------------------------------------------------------------------------- /test_img/20442421_1459247642.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/test_img/20442421_1459247642.jpg -------------------------------------------------------------------------------- /test_img/20442468_3479649590.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/test_img/20442468_3479649590.jpg -------------------------------------------------------------------------------- /test_img/20442515_3243295335.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/test_img/20442515_3243295335.jpg -------------------------------------------------------------------------------- /test_img/20442640_2409912250.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/test_img/20442640_2409912250.jpg -------------------------------------------------------------------------------- /test_img/20442750_205861976.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/test_img/20442750_205861976.jpg -------------------------------------------------------------------------------- /test_img/20442984_333319517.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/test_img/20442984_333319517.jpg -------------------------------------------------------------------------------- /test_img/md_img/attentionV2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/test_img/md_img/attentionV2.png -------------------------------------------------------------------------------- /test_img/md_img/attentionocr.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/test_img/md_img/attentionocr.png -------------------------------------------------------------------------------- /test_img/md_img/attention结果.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/test_img/md_img/attention结果.png -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenjun2hao/Attention_ocr.pytorch/HEAD/train.py --------------------------------------------------------------------------------