Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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 this issue Sep 19, 2022 · 1 comment

Comments

@aibohang
Copy link

请问在使用百度网盘提供的权重进行推理时出现权重加载不进去是什么原因?
{'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.linear_conv.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
Copy link
Owner

frotms commented Sep 29, 2022

@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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants