frotms

Results 122 comments of frotms

> > torch.jit.trace时需要修改DB网络输出,list改为tuple,dict改为NamedTuple。不然torch.jit.trace不支持 > > 请问具体是在哪里?base_model.py主网络构建里没有找到 db的head里

> > > rec的命令是? > > > > > > from torch.utils.mobile_optimizer import optimize_for_mobile example = torch.rand(1, 3, 640, 640) mobile = torch.jit.trace(self.net, example) traced_script_module_optimized = optimize_for_mobile(mobile) traced_script_module_optimized._save_for_lite_interpreter("ptocr_rec_v2.ptl") rec的命令也是如此 >...

用的dict是对的吗?或者检查下转换后的模型和原来提供的pth模型大小是否一致?或者方便提供一下模型吗

@libenchong 完整的命令和报错麻烦发出来看下。

@libenchong 由于这里是根据PaddleOCRv2.3改的,使用en模型命令: `python3 ./tools/infer/predict_rec.py --rec_model_path ./en_number_mobile_v2.0_rec_infer.pth --rec_char_dict_path ./pytorchocr/utils/dict/en_dict.txt --image_dir ./doc/imgs_words/en`

您确认一下,不需要指定rec_char_type。或者把完整截图发上来看看