frotms / PaddleOCR2Pytorch

PaddleOCR inference in PyTorch. Converted from [PaddleOCR](https://github.com/PaddlePaddle/PaddleOCR)
Apache License 2.0
809 stars 159 forks source link

python ./tools/infer/predict_det.py --det_algorithm DB --det_model_path ./weight/en_ptocr_v3_det_infer.pth --i mage_dir ./doc/imgs/1.jpg推理错误 #58

Open aibohang opened 1 year ago

aibohang commented 1 year ago

请问在使用百度网盘提供的权重进行推理时出现权重加载不进去是什么原因? {'fc_decay': 4e-05, 'in_channels': 96} Traceback (most recent call last): File "./tools/infer/predict_det.py", line 233, in text_detector = TextDetector(args) File "./tools/infer/predict_det.py", line 111, in init self.load_pytorch_weights(self.weights_path) File "F:\42_ZhuanPaddle\PaddleOCR2Pytorch-main\pytorchocr\base_ocr_v20.py", line 77, in load_pytorch_weights self.net.load_state_dict(torch.load(weights_path)) File "G:\Compiler\Anaconda2021.11\envs\cu102_torch110_py38_paddleocr\lib\site-packages\torch\nn\modules\module.py", line 1482, in load_state_dict raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format( RuntimeError: Error(s) in loading state_dict for BaseModel: Missing key(s) in state_dict: "backbone.conv1.conv.weight", "backbone.conv1.bn.weight", "backbone.conv1.bn.bias", "backbone.conv1.bn.running_mean", "backbone.conv1.bn.running_v ar", "backbone.blocks.0.expand_conv.conv.weight", "backbone.blocks.0.expand_conv.bn.weight", "backbone.blocks.0.expand_conv.bn.bias", "backbone.blocks.0.expand_conv.bn.running_mean", " backbone.blocks.0.expand_conv.bn.running_var", "backbone.blocks.0.bottleneck_conv.conv.weight", "backbone.blocks.0.bottleneck_conv.bn.weight", "backbone.blocks.0.bottleneck_conv.bn.bia s", "backbone.blocks.0.bottleneck_conv.bn.running_mean", "backbone.blocks.0.bottleneck_conv.bn.running_var", "backbone.blocks.0.mid_se.conv1.weight", "backbone.blocks.0.mid_se.conv1.bi as", "backbone.blocks.0.mid_se.conv2.weight", "backbone.blocks.0.mid_se.conv2.bias", "backbone.blocks.0.linear_conv.conv.weight", "backbone.blocks.0.linear_conv.bn.weight", "backbone.b locks.0.linear_conv.bn.bias", "backbone.blocks.0.linear_conv.bn.running_mean", "backbone.blocks.0.linear_conv.bn.running_var", "backbone.blocks.1.expand_conv.conv.weight", "backbone.bl ocks.1.expand_conv.bn.weight", "backbone.blocks.1.expand_conv.bn.bias", "backbone.blocks.1.expand_conv.bn.running_mean", "backbone.blocks.1.expand_conv.bn.running_var", "backbone.block s.1.bottleneck_conv.conv.weight", "backbone.blocks.1.bottleneck_conv.bn.weight", "backbone.blocks.1.bottleneck_conv.bn.bias", "backbone.blocks.1.bottleneck_conv.bn.running_mean", "back bone.blocks.1.bottleneck_conv.bn.running_var", "backbone.blocks.1.linear_conv.conv.weight", "backbone.blocks.1.linear_conv.bn.weight", "backbone.blocks.1.linear_conv.bn.bias", "backbon e.blocks.1.linear_conv.bn.running_mean", "backbone.blocks.1.linear_conv.bn.running_var", "backbone.blocks.2.expand_conv.conv.weight", "backbone.blocks.2.expand_conv.bn.weight", "backbo ne.blocks.2.expand_conv.bn.bias", "backbone.blocks.2.expand_conv.bn.running_mean", "backbone.blocks.2.expand_conv.bn.running_var", "backbone.blocks.2.bottleneck_conv.conv.weight", "bac kbone.blocks.2.bottleneck_conv.bn.weight", "backbone.blocks.2.bottleneck_conv.bn.bias", "backbone.blocks.2.bottleneck_conv.bn.running_mean", "backbone.blocks.2.bottleneck_conv.bn.runni ng_var", "backbone.blocks.2.linear_conv.conv.weight", "backbone.blocks.2.linear_conv.bn.weight", "backbone.blocks.2.linear_conv.bn.bias", "backbone.blocks.2.linear_conv.bn.running_mean ", "backbone.blocks.2.linear_conv.bn.running_var", "backbone.blocks.3.expand_conv.conv.weight", "backbone.blocks.3.expand_conv.bn.weight", "backbone.blocks.3.expand_conv.bn.bias", "bac kbone.blocks.3.expand_conv.bn.running_mean", "backbone.blocks.3.expand_conv.bn.running_var", "backbone.blocks.3.bottleneck_conv.conv.weight", "backbone.blocks.3.bottleneck_conv.bn.weig ht", "backbone.blocks.3.bottleneck_conv.bn.bias", "backbone.blocks.3.bottleneck_conv.bn.running_mean", "backbone.blocks.3.bottleneck_conv.bn.running_var", "backbone.blocks.3.mid_se.con v1.weight", "backbone.blocks.3.mid_se.conv1.bias", "backbone.blocks.3.mid_se.conv2.weight", "backbone.blocks.3.mid_se.conv2.bias", "backbone.blocks.3.linear_conv.conv.weight", "backbon e.blocks.3.linear_conv.bn.weight", "backbone.blocks.3.linear_conv.bn.bias", "backbone.blocks.3.linear_conv.bn.running_mean", "backbone.blocks.3.linear_conv.bn.running_var", "backbone.b locks.4.expand_conv.conv.weight", "backbone.blocks.4.expand_conv.bn.weight", "backbone.blocks.4.expand_conv.bn.bias", "backbone.blocks.4.expand_conv.bn.running_mean", "backbone.blocks. 4.expand_conv.bn.running_var", "backbone.blocks.4.bottleneck_conv.conv.weight", "backbone.blocks.4.bottleneck_conv.bn.weight", "backbone.blocks.4.bottleneck_conv.bn.bias", "backbone.bl ocks.4.bottleneck_conv.bn.running_mean", "backbone.blocks.4.bottleneck_conv.bn.running_var", "backbone.blocks.4.mid_se.conv1.weight", "backbone.blocks.4.mid_se.conv1.bias", "backbone.b locks.4.mid_se.conv2.weight", "backbone.blocks.4.mid_se.conv2.bias", "backbone.blocks.4.linear_conv.conv.weight", "backbone.blocks.4.linear_conv.bn.weight", "backbone.blocks.4.linear_c onv.bn.bias", "backbone.blocks.4.linear_conv.bn.running_mean", "backbone.blocks.4.linear_conv.bn.running_var", "backbone.blocks.5.expand_conv.conv.weight", "backbone.blocks.5.expand_co nv.bn.weight", "backbone.blocks.5.expand_conv.bn.bias", "backbone.blocks.5.expand_conv.bn.running_mean", "backbone.blocks.5.expand_conv.bn.running_var", "backbone.blocks.5.bottleneck_c onv.conv.weight", "backbone.blocks.5.bottleneck_conv.bn.weight", "backbone.blocks.5.bottleneck_conv.bn.bias", "backbone.blocks.5.bottleneck_conv.bn.running_mean", "backbone.blocks.5.bo ttleneck_conv.bn.running_var", "backbone.blocks.5.mid_se.conv1.weight", "backbone.blocks.5.mid_se.conv1.bias", "backbone.blocks.5.mid_se.conv2.weight", "backbone.blocks.5.mid_se.conv2. bias", "backbone.blocks.5.linear_conv.conv.weight", "backbone.blocks.5.linear_conv.bn.weight", "backbone.blocks.5.linear_conv.bn.bias", "backbone.blocks.5.linear_conv.bn.running_mean", "backbone.blocks.5.linear_conv.bn.running_var", "backbone.blocks.6.expand_conv.conv.weight", "backbone.blocks.6.expand_conv.bn.weight", "backbone.blocks.6.expand_conv.bn.bias", "backb one.blocks.6.expand_conv.bn.running_mean", "backbone.blocks.6.expand_conv.bn.running_var", "backbone.blocks.6.bottleneck_conv.conv.weight", "backbone.blocks.6.bottleneck_conv.bn.weight ", "backbone.blocks.6.bottleneck_conv.bn.bias", "backbone.blocks.6.bottleneck_conv.bn.running_mean", "backbone.blocks.6.bottleneck_conv.bn.running_var", "backbone.blocks.6.mid_se.conv1 .weight", "backbone.blocks.6.mid_se.conv1.bias", "backbone.blocks.6.mid_se.conv2.weight", "backbone.blocks.6.mid_se.conv2.bias", "backbone.blocks.6.linear_conv.conv.weight", "backbone. blocks.6.linear_conv.bn.weight", "backbone.blocks.6.linear_conv.bn.bias", "backbone.blocks.6.linear_conv.bn.running_mean", "backbone.blocks.6.linear_conv.bn.running_var", "backbone.blo cks.7.expand_conv.conv.weight", "backbone.blocks.7.expand_conv.bn.weight", "backbone.blocks.7.expand_conv.bn.bias", "backbone.blocks.7.expand_conv.bn.running_mean", "backbone.blocks.7. expand_conv.bn.running_var", "backbone.blocks.7.bottleneck_conv.conv.weight", "backbone.blocks.7.bottleneck_conv.bn.weight", "backbone.blocks.7.bottleneck_conv.bn.bias", "backbone.bloc ks.7.bottleneck_conv.bn.running_mean", "backbone.blocks.7.bottleneck_conv.bn.running_var", "backbone.blocks.7.mid_se.conv1.weight", "backbone.blocks.7.mid_se.conv1.bias", "backbone.blo cks.7.mid_se.conv2.weight", "backbone.blocks.7.mid_se.conv2.bias", "backbone.blocks.7.linear_conv.conv.weight", "backbone.blocks.7.linear_conv.bn.weight", "backbone.blocks.7.linear_con v.bn.bias", "backbone.blocks.7.linear_conv.bn.running_mean", "backbone.blocks.7.linear_conv.bn.running_var", "backbone.blocks.8.expand_conv.conv.weight", "backbone.blocks.8.expand_conv .bn.weight", "backbone.blocks.8.expand_conv.bn.bias", "backbone.blocks.8.expand_conv.bn.running_mean", "backbone.blocks.8.expand_conv.bn.running_var", "backbone.blocks.8.bottleneck_con v.conv.weight", "backbone.blocks.8.bottleneck_conv.bn.weight", "backbone.blocks.8.bottleneck_conv.bn.bias", "backbone.blocks.8.bottleneck_conv.bn.running_mean", "backbone.blocks.8.bott leneck_conv.bn.running_var", "backbone.blocks.8.mid_se.conv1.weight", "backbone.blocks.8.mid_se.conv1.bias", "backbone.blocks.8.mid_se.conv2.weight", "backbone.blocks.8.mid_se.conv2.bi as", "backbone.blocks.8.linear_conv.conv.weight", "backbone.blocks.8.linear_conv.bn.weight", "backbone.blocks.8.linear_conv.bn.bias", "backbone.blocks.8.linear_conv.bn.running_mean", " backbone.blocks.8.linear_conv.bn.running_var", "backbone.blocks.9.expand_conv.conv.weight", "backbone.blocks.9.expand_conv.bn.weight", "backbone.blocks.9.expand_conv.bn.bias", "backbon e.blocks.9.expand_conv.bn.running_mean", "backbone.blocks.9.expand_conv.bn.running_var", "backbone.blocks.9.bottleneck_conv.conv.weight", "backbone.blocks.9.bottleneck_conv.bn.weight", "backbone.blocks.9.bottleneck_conv.bn.bias", "backbone.blocks.9.bottleneck_conv.bn.running_mean", "backbone.blocks.9.bottleneck_conv.bn.running_var", "backbone.blocks.9.mid_se.conv1.w eight", "backbone.blocks.9.mid_se.conv1.bias", "backbone.blocks.9.mid_se.conv2.weight", "backbone.blocks.9.mid_se.conv2.bias", "backbone.blocks.9.linear_conv.conv.weight", "backbone.bl ocks.9.linear_conv.bn.weight", "backbone.blocks.9.linear_conv.bn.bias", "backbone.blocks.9.linear_conv.bn.running_mean", "backbone.blocks.9.linear_conv.bn.running_var", "backbone.block s.10.expand_conv.conv.weight", "backbone.blocks.10.expand_conv.bn.weight", "backbone.blocks.10.expand_conv.bn.bias", "backbone.blocks.10.expand_conv.bn.running_mean", "backbone.blocks. 10.expand_conv.bn.running_var", "backbone.blocks.10.bottleneck_conv.conv.weight", "backbone.blocks.10.bottleneck_conv.bn.weight", "backbone.blocks.10.bottleneck_conv.bn.bias", "backbon e.blocks.10.bottleneck_conv.bn.running_mean", "backbone.blocks.10.bottleneck_conv.bn.running_var", "backbone.blocks.10.mid_se.conv1.weight", "backbone.blocks.10.mid_se.conv1.bias", "ba ckbone.blocks.10.mid_se.conv2.weight", "backbone.blocks.10.mid_se.conv2.bias", "backbone.blocks.10.linear_conv.conv.weight", "backbone.blocks.10.linear_conv.bn.weight", "backbone.block s.10.linear_conv.bn.bias", "backbone.blocks.10.linear_conv.bn.running_mean", "backbone.blocks.10.linear_conv.bn.running_var", "backbone.conv2.conv.weight", "backbone.conv2.bn.weight", "backbone.conv2.bn.bias", "backbone.conv2.bn.running_mean", "backbone.conv2.bn.running_var", "neck.encoder.lstm.weight_ih_l0", "neck.encoder.lstm.weight_hh_l0", "neck.encoder.lstm.bias _ih_l0", "neck.encoder.lstm.bias_hh_l0", "neck.encoder.lstm.weight_ih_l0_reverse", "neck.encoder.lstm.weight_hh_l0_reverse", "neck.encoder.lstm.bias_ih_l0_reverse", "neck.encoder.lstm. bias_hh_l0_reverse", "neck.encoder.lstm.weight_ih_l1", "neck.encoder.lstm.weight_hh_l1", "neck.encoder.lstm.bias_ih_l1", "neck.encoder.lstm.bias_hh_l1", "neck.encoder.lstm.weight_ih_l1 _reverse", "neck.encoder.lstm.weight_hh_l1_reverse", "neck.encoder.lstm.bias_ih_l1_reverse", "neck.encoder.lstm.bias_hh_l1_reverse", "head.fc.weight", "head.fc.bias". Unexpected key(s) in state_dict: "backbone.conv.conv.weight", "backbone.conv.bn.weight", "backbone.conv.bn.bias", "backbone.conv.bn.running_mean", "backbone.conv.bn.running_var ", "backbone.conv.bn.num_batches_tracked", "backbone.stages.0.0.expand_conv.conv.weight", "backbone.stages.0.0.expand_conv.bn.weight", "backbone.stages.0.0.expand_conv.bn.bias", "backb one.stages.0.0.expand_conv.bn.running_mean", "backbone.stages.0.0.expand_conv.bn.running_var", "backbone.stages.0.0.expand_conv.bn.num_batches_tracked", "backbone.stages.0.0.bottleneck _conv.conv.weight", "backbone.stages.0.0.bottleneck_conv.bn.weight", "backbone.stages.0.0.bottleneck_conv.bn.bias", "backbone.stages.0.0.bottleneck_conv.bn.running_mean", "backbone.sta ges.0.0.bottleneck_conv.bn.running_var", "backbone.stages.0.0.bottleneck_conv.bn.num_batches_tracked", "backbone.stages.0.0.linear_conv.conv.weight", "backbone.stages.0.0.linear_conv.b n.weight", "backbone.stages.0.0.linear_conv.bn.bias", "backbone.stages.0.0.linear_conv.bn.running_mean", "backbone.stages.0.0.linear_conv.bn.running_var", "backbone.stages.0.0.linear_c onv.bn.num_batches_tracked", "backbone.stages.0.1.expand_conv.conv.weight", "backbone.stages.0.1.expand_conv.bn.weight", "backbone.stages.0.1.expand_conv.bn.bias", "backbone.stages.0.1 .expand_conv.bn.running_mean", "backbone.stages.0.1.expand_conv.bn.running_var", "backbone.stages.0.1.expand_conv.bn.num_batches_tracked", "backbone.stages.0.1.bottleneck_conv.conv.wei ght", "backbone.stages.0.1.bottleneck_conv.bn.weight", "backbone.stages.0.1.bottleneck_conv.bn.bias", "backbone.stages.0.1.bottleneck_conv.bn.running_mean", "backbone.stages.0.1.bottle neck_conv.bn.running_var", "backbone.stages.0.1.bottleneck_conv.bn.num_batches_tracked", "backbone.stages.0.1.linear_conv.conv.weight", "backbone.stages.0.1.linear_conv.bn.weight", "ba ckbone.stages.0.1.linear_conv.bn.bias", "backbone.stages.0.1.linear_conv.bn.running_mean", "backbone.stages.0.1.linear_conv.bn.running_var", "backbone.stages.0.1.linear_conv.bn.num_bat ches_tracked", "backbone.stages.0.2.expand_conv.conv.weight", "backbone.stages.0.2.expand_conv.bn.weight", "backbone.stages.0.2.expand_conv.bn.bias", "backbone.stages.0.2.expand_conv.b n.running_mean", "backbone.stages.0.2.expand_conv.bn.running_var", "backbone.stages.0.2.expand_conv.bn.num_batches_tracked", "backbone.stages.0.2.bottleneck_conv.conv.weight", "backbon e.stages.0.2.bottleneck_conv.bn.weight", "backbone.stages.0.2.bottleneck_conv.bn.bias", "backbone.stages.0.2.bottleneck_conv.bn.running_mean", "backbone.stages.0.2.bottleneck_conv.bn.r unning_var", "backbone.stages.0.2.bottleneck_conv.bn.num_batches_tracked", "backbone.stages.0.2.linear_conv.conv.weight", "backbone.stages.0.2.linear_conv.bn.weight", "backbone.stages. 0.2.linear_conv.bn.bias", "backbone.stages.0.2.linear_conv.bn.running_mean", "backbone.stages.0.2.linear_conv.bn.running_var", "backbone.stages.0.2.linear_conv.bn.num_batches_tracked", "backbone.stages.1.0.expand_conv.conv.weight", "backbone.stages.1.0.expand_conv.bn.weight", "backbone.stages.1.0.expand_conv.bn.bias", "backbone.stages.1.0.expand_conv.bn.running_mean ", "backbone.stages.1.0.expand_conv.bn.running_var", "backbone.stages.1.0.expand_conv.bn.num_batches_tracked", "backbone.stages.1.0.bottleneck_conv.conv.weight", "backbone.stages.1.0.b ottleneck_conv.bn.weight", "backbone.stages.1.0.bottleneck_conv.bn.bias", "backbone.stages.1.0.bottleneck_conv.bn.running_mean", "backbone.stages.1.0.bottleneck_conv.bn.running_var", " backbone.stages.1.0.bottleneck_conv.bn.num_batches_tracked", "backbone.stages.1.0.linear_conv.conv.weight", "backbone.stages.1.0.linear_conv.bn.weight", "backbone.stages.1.0.linear_con v.bn.bias", "backbone.stages.1.0.linear_conv.bn.running_mean", "backbone.stages.1.0.linear_conv.bn.running_var", "backbone.stages.1.0.linear_conv.bn.num_batches_tracked", "backbone.sta ges.1.1.expand_conv.conv.weight", "backbone.stages.1.1.expand_conv.bn.weight", "backbone.stages.1.1.expand_conv.bn.bias", "backbone.stages.1.1.expand_conv.bn.running_mean", "backbone.s tages.1.1.expand_conv.bn.running_var", "backbone.stages.1.1.expand_conv.bn.num_batches_tracked", "backbone.stages.1.1.bottleneck_conv.conv.weight", "backbone.stages.1.1.bottleneck_conv .bn.weight", "backbone.stages.1.1.bottleneck_conv.bn.bias", "backbone.stages.1.1.bottleneck_conv.bn.running_mean", "backbone.stages.1.1.bottleneck_conv.bn.running_var", "backbone.stage s.1.1.bottleneck_conv.bn.num_batches_tracked", "backbone.stages.1.1.linear_conv.conv.weight", "backbone.stages.1.1.linear_conv.bn.weight", "backbone.stages.1.1.linear_conv.bn.bias", "b ackbone.stages.1.1.linear_conv.bn.running_mean", "backbone.stages.1.1.linear_conv.bn.running_var", "backbone.stages.1.1.linear_conv.bn.num_batches_tracked", "backbone.stages.1.2.expand _conv.conv.weight", "backbone.stages.1.2.expand_conv.bn.weight", "backbone.stages.1.2.expand_conv.bn.bias", "backbone.stages.1.2.expand_conv.bn.running_mean", "backbone.stages.1.2.expa nd_conv.bn.running_var", "backbone.stages.1.2.expand_conv.bn.num_batches_tracked", "backbone.stages.1.2.bottleneck_conv.conv.weight", "backbone.stages.1.2.bottleneck_conv.bn.weight", " backbone.stages.1.2.bottleneck_conv.bn.bias", "backbone.stages.1.2.bottleneck_conv.bn.running_mean", "backbone.stages.1.2.bottleneck_conv.bn.running_var", "backbone.stages.1.2.bottlene ck_conv.bn.num_batches_tracked", "backbone.stages.1.2.linear_conv.conv.weight", "backbone.stages.1.2.linear_conv.bn.weight", "backbone.stages.1.2.linear_conv.bn.bias", "backbone.stages .1.2.linear_conv.bn.running_mean", "backbone.stages.1.2.linear_conv.bn.running_var", "backbone.stages.1.2.linear_conv.bn.num_batches_tracked", "backbone.stages.2.0.expand_conv.conv.wei ght", "backbone.stages.2.0.expand_conv.bn.weight", "backbone.stages.2.0.expand_conv.bn.bias", "backbone.stages.2.0.expand_conv.bn.running_mean", "backbone.stages.2.0.expand_conv.bn.run ning_var", "backbone.stages.2.0.expand_conv.bn.num_batches_tracked", "backbone.stages.2.0.bottleneck_conv.conv.weight", "backbone.stages.2.0.bottleneck_conv.bn.weight", "backbone.stage s.2.0.bottleneck_conv.bn.bias", "backbone.stages.2.0.bottleneck_conv.bn.running_mean", "backbone.stages.2.0.bottleneck_conv.bn.running_var", "backbone.stages.2.0.bottleneck_conv.bn.num _batches_tracked", "backbone.stages.2.0.linear_conv.conv.weight", "backbone.stages.2.0.linear_conv.bn.weight", "backbone.stages.2.0.linear_conv.bn.bias", "backbone.stages.2.0.linear_co nv.bn.running_mean", "backbone.stages.2.0.linear_conv.bn.running_var", "backbone.stages.2.0.linear_conv.bn.num_batches_tracked", "backbone.stages.2.1.expand_conv.conv.weight", "backbon e.stages.2.1.expand_conv.bn.weight", "backbone.stages.2.1.expand_conv.bn.bias", "backbone.stages.2.1.expand_conv.bn.running_mean", "backbone.stages.2.1.expand_conv.bn.running_var", "ba ckbone.stages.2.1.expand_conv.bn.num_batches_tracked", "backbone.stages.2.1.bottleneck_conv.conv.weight", "backbone.stages.2.1.bottleneck_conv.bn.weight", "backbone.stages.2.1.bottlene ck_conv.bn.bias", "backbone.stages.2.1.bottleneck_conv.bn.running_mean", "backbone.stages.2.1.bottleneck_conv.bn.running_var", "backbone.stages.2.1.bottleneck_conv.bn.num_batches_track ed", "backbone.stages.2.1.linear_conv.conv.weight", "backbone.stages.2.1.linear_conv.bn.weight", "backbone.stages.2.1.linear_conv.bn.bias", "backbone.stages.2.1.linearconv.bn.running mean", "backbone.stages.2.1.linear_conv.bn.running_var", "backbone.stages.2.1.linear_conv.bn.num_batches_tracked", "backbone.stages.2.2.expand_conv.conv.weight", "backbone.stages.2.2.e xpand_conv.bn.weight", "backbone.stages.2.2.expand_conv.bn.bias", "backbone.stages.2.2.expand_conv.bn.running_mean", "backbone.stages.2.2.expand_conv.bn.running_var", "backbone.stages. 2.2.expand_conv.bn.num_batches_tracked", "backbone.stages.2.2.bottleneck_conv.conv.weight", "backbone.stages.2.2.bottleneck_conv.bn.weight", "backbone.stages.2.2.bottleneck_conv.bn.bia s", "backbone.stages.2.2.bottleneck_conv.bn.running_mean", "backbone.stages.2.2.bottleneck_conv.bn.running_var", "backbone.stages.2.2.bottleneck_conv.bn.num_batches_tracked", "backbone .stages.2.2.linear_conv.conv.weight", "backbone.stages.2.2.linear_conv.bn.weight", "backbone.stages.2.2.linear_conv.bn.bias", "backbone.stages.2.2.linear_conv.bn.running_mean", "backbo ne.stages.2.2.linear_conv.bn.running_var", "backbone.stages.2.2.linear_conv.bn.num_batches_tracked", "backbone.stages.2.3.expand_conv.conv.weight", "backbone.stages.2.3.expand_conv.bn. weight", "backbone.stages.2.3.expand_conv.bn.bias", "backbone.stages.2.3.expand_conv.bn.running_mean", "backbone.stages.2.3.expand_conv.bn.running_var", "backbone.stages.2.3.expand_con v.bn.num_batches_tracked", "backbone.stages.2.3.bottleneck_conv.conv.weight", "backbone.stages.2.3.bottleneck_conv.bn.weight", "backbone.stages.2.3.bottleneck_conv.bn.bias", "backbone. stages.2.3.bottleneck_conv.bn.running_mean", "backbone.stages.2.3.bottleneck_conv.bn.running_var", "backbone.stages.2.3.bottleneck_conv.bn.num_batches_tracked", "backbone.stages.2.3.li near_conv.conv.weight", "backbone.stages.2.3.linear_conv.bn.weight", "backbone.stages.2.3.linear_conv.bn.bias", "backbone.stages.2.3.linear_conv.bn.running_mean", "backbone.stages.2.3. linear_conv.bn.running_var", "backbone.stages.2.3.linear_conv.bn.num_batches_tracked", "backbone.stages.2.4.expand_conv.conv.weight", "backbone.stages.2.4.expand_conv.bn.weight", "back bone.stages.2.4.expand_conv.bn.bias", "backbone.stages.2.4.expand_conv.bn.running_mean", "backbone.stages.2.4.expand_conv.bn.running_var", "backbone.stages.2.4.expand_conv.bn.num_batch es_tracked", "backbone.stages.2.4.bottleneck_conv.conv.weight", "backbone.stages.2.4.bottleneck_conv.bn.weight", "backbone.stages.2.4.bottleneck_conv.bn.bias", "backbone.stages.2.4.bot tleneck_conv.bn.running_mean", "backbone.stages.2.4.bottleneck_conv.bn.running_var", "backbone.stages.2.4.bottleneck_conv.bn.num_batches_tracked", "backbone.stages.2.4.linear_conv.conv .weight", "backbone.stages.2.4.linear_conv.bn.weight", "backbone.stages.2.4.linear_conv.bn.bias", "backbone.stages.2.4.linear_conv.bn.running_mean", "backbone.stages.2.4.linear_conv.bn .running_var", "backbone.stages.2.4.linear_conv.bn.num_batches_tracked", "backbone.stages.2.5.expand_conv.conv.weight", "backbone.stages.2.5.expand_conv.bn.weight", "backbone.stages.2. 5.expand_conv.bn.bias", "backbone.stages.2.5.expand_conv.bn.running_mean", "backbone.stages.2.5.expand_conv.bn.running_var", "backbone.stages.2.5.expand_conv.bn.num_batches_tracked", " backbone.stages.2.5.bottleneck_conv.conv.weight", "backbone.stages.2.5.bottleneck_conv.bn.weight", "backbone.stages.2.5.bottleneck_conv.bn.bias", "backbone.stages.2.5.bottleneck_conv.b n.running_mean", "backbone.stages.2.5.bottleneck_conv.bn.running_var", "backbone.stages.2.5.bottleneck_conv.bn.num_batches_tracked", "backbone.stages.2.5.linear_conv.conv.weight", "bac kbone.stages.2.5.linear_conv.bn.weight", "backbone.stages.2.5.linear_conv.bn.bias", "backbone.stages.2.5.linear_conv.bn.running_mean", "backbone.stages.2.5.linear_conv.bn.running_var", "backbone.stages.2.5.linear_conv.bn.num_batches_tracked", "backbone.stages.3.0.expand_conv.conv.weight", "backbone.stages.3.0.expand_conv.bn.weight", "backbone.stages.3.0.expand_conv. bn.bias", "backbone.stages.3.0.expand_conv.bn.running_mean", "backbone.stages.3.0.expand_conv.bn.running_var", "backbone.stages.3.0.expand_conv.bn.num_batches_tracked", "backbone.stage s.3.0.bottleneck_conv.conv.weight", "backbone.stages.3.0.bottleneck_conv.bn.weight", "backbone.stages.3.0.bottleneck_conv.bn.bias", "backbone.stages.3.0.bottleneck_conv.bn.running_mean ", "backbone.stages.3.0.bottleneck_conv.bn.running_var", "backbone.stages.3.0.bottleneck_conv.bn.num_batches_tracked", "backbone.stages.3.0.linear_conv.conv.weight", "backbone.stages.3 .0.linear_conv.bn.weight", "backbone.stages.3.0.linear_conv.bn.bias", "backbone.stages.3.0.linear_conv.bn.running_mean", "backbone.stages.3.0.linear_conv.bn.running_var", "backbone.sta ges.3.0.linear_conv.bn.num_batches_tracked", "backbone.stages.3.1.expand_conv.conv.weight", "backbone.stages.3.1.expand_conv.bn.weight", "backbone.stages.3.1.expand_conv.bn.bias", "bac kbone.stages.3.1.expand_conv.bn.running_mean", "backbone.stages.3.1.expand_conv.bn.running_var", "backbone.stages.3.1.expand_conv.bn.num_batches_tracked", "backbone.stages.3.1.bottlene ck_conv.conv.weight", "backbone.stages.3.1.bottleneck_conv.bn.weight", "backbone.stages.3.1.bottleneck_conv.bn.bias", "backbone.stages.3.1.bottleneck_conv.bn.running_mean", "backbone.s tages.3.1.bottleneck_conv.bn.running_var", "backbone.stages.3.1.bottleneck_conv.bn.num_batches_tracked", "backbone.stages.3.1.linear_conv.conv.weight", "backbone.stages.3.1.linear_conv .bn.weight", "backbone.stages.3.1.linear_conv.bn.bias", "backbone.stages.3.1.linear_conv.bn.running_mean", "backbone.stages.3.1.linear_conv.bn.running_var", "backbone.stages.3.1.linear _conv.bn.num_batches_tracked", "backbone.stages.3.2.expand_conv.conv.weight", "backbone.stages.3.2.expand_conv.bn.weight", "backbone.stages.3.2.expand_conv.bn.bias", "backbone.stages.3 .2.expand_conv.bn.running_mean", "backbone.stages.3.2.expand_conv.bn.running_var", "backbone.stages.3.2.expand_conv.bn.num_batches_tracked", "backbone.stages.3.2.bottleneck_conv.conv.w eight", "backbone.stages.3.2.bottleneck_conv.bn.weight", "backbone.stages.3.2.bottleneck_conv.bn.bias", "backbone.stages.3.2.bottleneck_conv.bn.running_mean", "backbone.stages.3.2.bott leneck_conv.bn.running_var", "backbone.stages.3.2.bottleneck_conv.bn.num_batches_tracked", "backbone.stages.3.2.linear_conv.conv.weight", "backbone.stages.3.2.linear_conv.bn.weight", " backbone.stages.3.2.linear_conv.bn.bias", "backbone.stages.3.2.linear_conv.bn.running_mean", "backbone.stages.3.2.linear_conv.bn.running_var", "backbone.stages.3.2.linear_conv.bn.num_b atches_tracked", "backbone.stages.3.3.conv.weight", "backbone.stages.3.3.bn.weight", "backbone.stages.3.3.bn.bias", "backbone.stages.3.3.bn.running_mean", "backbone.stages.3.3.bn.runni ng_var", "backbone.stages.3.3.bn.num_batches_tracked", "neck.ins_conv.0.in_conv.weight", "neck.ins_conv.0.se_block.conv1.weight", "neck.ins_conv.0.se_block.conv1.bias", "neck.ins_conv. 0.se_block.conv2.weight", "neck.ins_conv.0.se_block.conv2.bias", "neck.ins_conv.1.in_conv.weight", "neck.ins_conv.1.se_block.conv1.weight", "neck.ins_conv.1.se_block.conv1.bias", "neck .ins_conv.1.se_block.conv2.weight", "neck.ins_conv.1.se_block.conv2.bias", "neck.ins_conv.2.in_conv.weight", "neck.ins_conv.2.se_block.conv1.weight", "neck.ins_conv.2.se_block.conv1.bi as", "neck.ins_conv.2.se_block.conv2.weight", "neck.ins_conv.2.se_block.conv2.bias", "neck.ins_conv.3.in_conv.weight", "neck.ins_conv.3.se_block.conv1.weight", "neck.ins_conv.3.se_bloc k.conv1.bias", "neck.ins_conv.3.se_block.conv2.weight", "neck.ins_conv.3.se_block.conv2.bias", "neck.inp_conv.0.in_conv.weight", "neck.inp_conv.0.se_block.conv1.weight", "neck.inp_conv .0.se_block.conv1.bias", "neck.inp_conv.0.se_block.conv2.weight", "neck.inp_conv.0.se_block.conv2.bias", "neck.inp_conv.1.in_conv.weight", "neck.inp_conv.1.se_block.conv1.weight", "nec k.inp_conv.1.se_block.conv1.bias", "neck.inp_conv.1.se_block.conv2.weight", "neck.inp_conv.1.se_block.conv2.bias", "neck.inp_conv.2.in_conv.weight", "neck.inp_conv.2.se_block.conv1.wei ght", "neck.inp_conv.2.se_block.conv1.bias", "neck.inp_conv.2.se_block.conv2.weight", "neck.inp_conv.2.se_block.conv2.bias", "neck.inp_conv.3.in_conv.weight", "neck.inp_conv.3.se_block .conv1.weight", "neck.inp_conv.3.se_block.conv1.bias", "neck.inp_conv.3.se_block.conv2.weight", "neck.inp_conv.3.se_block.conv2.bias", "head.binarize.conv1.weight", "head.binarize.conv _bn1.weight", "head.binarize.conv_bn1.bias", "head.binarize.conv_bn1.running_mean", "head.binarize.conv_bn1.running_var", "head.binarize.conv_bn1.num_batches_tracked", "head.binarize.c onv2.weight", "head.binarize.conv2.bias", "head.binarize.conv_bn2.weight", "head.binarize.conv_bn2.bias", "head.binarize.conv_bn2.running_mean", "head.binarize.conv_bn2.running_var", " head.binarize.conv_bn2.num_batches_tracked", "head.binarize.conv3.weight", "head.binarize.conv3.bias", "head.thresh.conv1.weight", "head.thresh.conv_bn1.weight", "head.thresh.conv_bn1. bias", "head.thresh.conv_bn1.running_mean", "head.thresh.conv_bn1.running_var", "head.thresh.conv_bn1.num_batches_tracked", "head.thresh.conv2.weight", "head.thresh.conv2.bias", "head. thresh.conv_bn2.weight", "head.thresh.conv_bn2.bias", "head.thresh.conv_bn2.running_mean", "head.thresh.conv_bn2.running_var", "head.thresh.conv_bn2.num_batches_tracked", "head.thresh. conv3.weight", "head.thresh.conv3.bias".

frotms commented 1 year ago

@aibohang 请使用 python ./tools/infer/predict_det.py --det_algorithm DB --det_yaml_path ./configs/det/det_ppocr_v3.yml --det_model_path ./weight/en_ptocr_v3_det_infer.pth --i mage_dir ./doc/imgs/1.jpg