├── LICENSE ├── README.md ├── alphabet.py ├── dataloader_syn.py ├── dataset.py ├── eval.py ├── figs └── STKM.png ├── img_test.jpg ├── loss.py ├── model ├── resnet.py └── transformers.py ├── requirements.txt ├── train.py └── trans2res.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVI-SZU/STKM/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVI-SZU/STKM/HEAD/README.md -------------------------------------------------------------------------------- /alphabet.py: -------------------------------------------------------------------------------- 1 | alphabet = """!\"#$%\'()*+,-./0123456789:;<=>?@ABCDEFGHIJKLMNOPQRSTUVWXY[\\]^_`abcdefghijklmnopqrstuvwxyz{|}~""" -------------------------------------------------------------------------------- /dataloader_syn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVI-SZU/STKM/HEAD/dataloader_syn.py -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVI-SZU/STKM/HEAD/dataset.py -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVI-SZU/STKM/HEAD/eval.py -------------------------------------------------------------------------------- /figs/STKM.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVI-SZU/STKM/HEAD/figs/STKM.png -------------------------------------------------------------------------------- /img_test.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVI-SZU/STKM/HEAD/img_test.jpg -------------------------------------------------------------------------------- /loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVI-SZU/STKM/HEAD/loss.py -------------------------------------------------------------------------------- /model/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVI-SZU/STKM/HEAD/model/resnet.py -------------------------------------------------------------------------------- /model/transformers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVI-SZU/STKM/HEAD/model/transformers.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | numpy==1.21.5 2 | opencv_python==4.5.5.64 3 | Pillow==9.2.0 4 | -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVI-SZU/STKM/HEAD/train.py -------------------------------------------------------------------------------- /trans2res.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVI-SZU/STKM/HEAD/trans2res.py --------------------------------------------------------------------------------