wenwenyu / MASTER-pytorch

Code for the paper "MASTER: Multi-Aspect Non-local Network for Scene Text Recognition" (Pattern Recognition 2021)
https://arxiv.org/abs/1910.02562
MIT License
277 stars 53 forks source link

label编码的代码在哪 #11

Open renmada opened 3 years ago

renmada commented 3 years ago

dataset里的代码是
if self.training: if not self.case_sensitive: label = label.lower() return (img, label) DistCollateFn里没有看到相关代码

输入给模型的label不能是str吧

zezeze97 commented 3 years ago

在utils/label_util.py里