shannanyinxiang / SPTS

Official implementation of SPTS: Single-Point Text Spotting (ACM MM 2022 Oral)
137 stars 12 forks source link

请问如何进行无坐标的训练? #3

Closed Airwithcode closed 1 year ago

Airwithcode commented 1 year ago

需要修改哪一块的代码

shannanyinxiang commented 1 year ago

修改dataset/collate_fn.py中的SeqConstructor类,使其在构建序列时仅包括文本序列。

具体地,可以将dataset/collate_fn.py中43行的pt_label = torch.cat([center_pts, recog_label], dim=-1)修改为pt_label = recog_label

Airwithcode commented 1 year ago

感谢您的回答,训练完成后,推理没有结果,请问需要修改哪部分代码

shannanyinxiang commented 1 year ago

修改engine/val.py中的decode_pred_seq函数,让它只解码文本。还需要修改model/transformer.py的71行到79行。