search

mxin262 / SwinTextSpotter

Pytorch re-implementation of Paper: SwinTextSpotter: Scene Text Spotting via Better Synergy between Text Detection and Text Recognition (CVPR 2022)
272 stars 42 forks source link

作者你好，初学者想询问一个问题 #98

Open ssygeneral opened 1 year ago

ssygeneral commented 1 year ago

请问，我在使用您已经训练好的模型进行测试的时候，发现它只能识别英文或者中文。如果我想识别英文，中文以及数字和一些字符都在一张图片上的照片应该做哪些修改才能实现

mxin262 commented 1 year ago

使用ReCTS上训练的模型应该可以同时识别英文和中文

ssygeneral commented 1 year ago

ReCTS模型确实可以同时识别英文和中文，但是我对它进行了微调，使用totaltext在rects的模型权重上进行了再训练，得到的模型却识别出来的却是各种中文乱码，这是为什么？下面是配置文件。 BASE: "Base-SWINTS_swin.yaml" MODEL: WEIGHTS: "/swin-transformer/SwinTextSpotter-main/output/rects_model_final.pth" SWINTS: NUM_PROPOSALS: 300 NUM_CLASSES: 2 REC_HEAD: POOLER_RESOLUTION: (16,40) RESOLUTION: (32, 80) BATCH_SIZE: 128 NUM_CLASSES: 5463 DATASETS:

TRAIN: ("rects",) TEST: ("totaltext_test",) SOLVER: STEPS: (140000,160000) MAX_ITER: 20000 CHECKPOINT_PERIOD: 5000 INPUT: FORMAT: "RGB"

mxin262 commented 1 year ago

Total-Text和ReCTS识别的码表不同