在运行LORE模型的demo_wireless.sh时,运行结果出错。
demo_wireless.sh
python demo.py ctdet_mid \ --dataset table_mid \ --demo ../input_images/wireless \ --demo_name demo_wireless \ --debug 1 \ --arch resfpnhalf_18 \ --K 3000 \ --MK 5000 \ --upper_left \ --tsfm_layers 4\ --stacking_layers 4 \ --gpus 0\ --wiz_2dpe \ --wiz_detect \ --wiz_stacking \ --convert_onnx 0 \ --vis_thresh_corner 0.3 \ --vis_thresh 0.2 \ --scores_thresh 0.2 \ --nms \ --demo_dir ../output/test_wireless/ \ --load_model ../dir_of_ckpt/ckpt_wireless/model_best.pth \ --load_processor ../dir_of_ckpt/ckpt_wireless/processor_best.pth
预训练模型能够正常加载,但是存在一些问题,
Fix size testing. training chunk_sizes: [1] The output will be saved to /root/data1/bwq/LORE_new/AdvancedLiterateMachinery/DocumentUnderstanding/LORE-TSR/src/lib/../../exp/ctdet_mid/default heads {'hm': 2, 'st': 8, 'wh': 8, 'ax': 256, 'cr': 256, 'reg': 2} loaded ../dir_of_ckpt/ckpt_wireless/model_best.pth, epoch 100 Drop parameter base.base_layer.0.weight. Drop parameter base.base_layer.1.weight. Drop parameter base.base_layer.1.bias. Drop parameter base.base_layer.1.running_mean. Drop parameter base.base_layer.1.running_var. Drop parameter base.base_layer.1.num_batches_tracked. Drop parameter base.level0.0.weight. Drop parameter base.level0.1.weight. Drop parameter base.level0.1.bias. Drop parameter base.level0.1.running_mean. Drop parameter base.level0.1.running_var. Drop parameter base.level0.1.num_batches_tracked. Drop parameter base.level1.0.weight. Drop parameter base.level1.1.weight. Drop parameter base.level1.1.bias. Drop parameter base.level1.1.running_mean. Drop parameter base.level1.1.running_var. Drop parameter base.level1.1.num_batches_tracked. Drop parameter base.level2.tree1.conv1.weight. Drop parameter base.level2.tree1.bn1.weight. Drop parameter base.level2.tree1.bn1.bias. Drop parameter base.level2.tree1.bn1.running_mean. Drop parameter base.level2.tree1.bn1.running_var. Drop parameter base.level2.tree1.bn1.num_batches_tracked. Drop parameter base.level2.tree1.conv2.weight. Drop parameter base.level2.tree1.bn2.weight. Drop parameter base.level2.tree1.bn2.bias. Drop parameter base.level2.tree1.bn2.running_mean. Drop parameter base.level2.tree1.bn2.running_var. Drop parameter base.level2.tree1.bn2.num_batches_tracked. Drop parameter base.level2.tree2.conv1.weight. Drop parameter base.level2.tree2.bn1.weight. Drop parameter base.level2.tree2.bn1.bias. Drop parameter base.level2.tree2.bn1.running_mean. Drop parameter base.level2.tree2.bn1.running_var. Drop parameter base.level2.tree2.bn1.num_batches_tracked. Drop parameter base.level2.tree2.conv2.weight. Drop parameter base.level2.tree2.bn2.weight. Drop parameter base.level2.tree2.bn2.bias. Drop parameter base.level2.tree2.bn2.running_mean. Drop parameter base.level2.tree2.bn2.running_var. Drop parameter base.level2.tree2.bn2.num_batches_tracked. Drop parameter base.level2.root.conv.weight. Drop parameter base.level2.root.bn.weight. Drop parameter base.level2.root.bn.bias. Drop parameter base.level2.root.bn.running_mean. Drop parameter base.level2.root.bn.running_var. Drop parameter base.level2.root.bn.num_batches_tracked. Drop parameter base.level2.project.0.weight. Drop parameter base.level2.project.1.weight. Drop parameter base.level2.project.1.bias. Drop parameter base.level2.project.1.running_mean. Drop parameter base.level2.project.1.running_var. Drop parameter base.level2.project.1.num_batches_tracked. Drop parameter base.level3.tree1.tree1.conv1.weight. Drop parameter base.level3.tree1.tree1.bn1.weight. Drop parameter base.level3.tree1.tree1.bn1.bias. Drop parameter base.level3.tree1.tree1.bn1.running_mean. Drop parameter base.level3.tree1.tree1.bn1.running_var. Drop parameter base.level3.tree1.tree1.bn1.num_batches_tracked. Drop parameter base.level3.tree1.tree1.conv2.weight. Drop parameter base.level3.tree1.tree1.bn2.weight. Drop parameter base.level3.tree1.tree1.bn2.bias. Drop parameter base.level3.tree1.tree1.bn2.running_mean. Drop parameter base.level3.tree1.tree1.bn2.running_var. Drop parameter base.level3.tree1.tree1.bn2.num_batches_tracked. Drop parameter base.level3.tree1.tree2.conv1.weight. Drop parameter base.level3.tree1.tree2.bn1.weight. Drop parameter base.level3.tree1.tree2.bn1.bias. Drop parameter base.level3.tree1.tree2.bn1.running_mean. Drop parameter base.level3.tree1.tree2.bn1.running_var. Drop parameter base.level3.tree1.tree2.bn1.num_batches_tracked. Drop parameter base.level3.tree1.tree2.conv2.weight. Drop parameter base.level3.tree1.tree2.bn2.weight. Drop parameter base.level3.tree1.tree2.bn2.bias. Drop parameter base.level3.tree1.tree2.bn2.running_mean. Drop parameter base.level3.tree1.tree2.bn2.running_var. Drop parameter base.level3.tree1.tree2.bn2.num_batches_tracked. Drop parameter base.level3.tree1.root.conv.weight. Drop parameter base.level3.tree1.root.bn.weight. Drop parameter base.level3.tree1.root.bn.bias. Drop parameter base.level3.tree1.root.bn.running_mean. Drop parameter base.level3.tree1.root.bn.running_var. Drop parameter base.level3.tree1.root.bn.num_batches_tracked. Drop parameter base.level3.tree1.project.0.weight. Drop parameter base.level3.tree1.project.1.weight. Drop parameter base.level3.tree1.project.1.bias. Drop parameter base.level3.tree1.project.1.running_mean. Drop parameter base.level3.tree1.project.1.running_var. Drop parameter base.level3.tree1.project.1.num_batches_tracked. Drop parameter base.level3.tree2.tree1.conv1.weight. Drop parameter base.level3.tree2.tree1.bn1.weight. Drop parameter base.level3.tree2.tree1.bn1.bias. Drop parameter base.level3.tree2.tree1.bn1.running_mean. Drop parameter base.level3.tree2.tree1.bn1.running_var. Drop parameter base.level3.tree2.tree1.bn1.num_batches_tracked. Drop parameter base.level3.tree2.tree1.conv2.weight. Drop parameter base.level3.tree2.tree1.bn2.weight. Drop parameter base.level3.tree2.tree1.bn2.bias. Drop parameter base.level3.tree2.tree1.bn2.running_mean. Drop parameter base.level3.tree2.tree1.bn2.running_var. Drop parameter base.level3.tree2.tree1.bn2.num_batches_tracked. Drop parameter base.level3.tree2.tree2.conv1.weight. Drop parameter base.level3.tree2.tree2.bn1.weight. Drop parameter base.level3.tree2.tree2.bn1.bias. Drop parameter base.level3.tree2.tree2.bn1.running_mean. Drop parameter base.level3.tree2.tree2.bn1.running_var. Drop parameter base.level3.tree2.tree2.bn1.num_batches_tracked. Drop parameter base.level3.tree2.tree2.conv2.weight. Drop parameter base.level3.tree2.tree2.bn2.weight. Drop parameter base.level3.tree2.tree2.bn2.bias. Drop parameter base.level3.tree2.tree2.bn2.running_mean. Drop parameter base.level3.tree2.tree2.bn2.running_var. Drop parameter base.level3.tree2.tree2.bn2.num_batches_tracked. Drop parameter base.level3.tree2.root.conv.weight. Drop parameter base.level3.tree2.root.bn.weight. Drop parameter base.level3.tree2.root.bn.bias. Drop parameter base.level3.tree2.root.bn.running_mean. Drop parameter base.level3.tree2.root.bn.running_var. Drop parameter base.level3.tree2.root.bn.num_batches_tracked. Drop parameter base.level3.project.0.weight. Drop parameter base.level3.project.1.weight. Drop parameter base.level3.project.1.bias. Drop parameter base.level3.project.1.running_mean. Drop parameter base.level3.project.1.running_var. Drop parameter base.level3.project.1.num_batches_tracked. Drop parameter base.level4.tree1.tree1.conv1.weight. Drop parameter base.level4.tree1.tree1.bn1.weight. Drop parameter base.level4.tree1.tree1.bn1.bias. Drop parameter base.level4.tree1.tree1.bn1.running_mean. Drop parameter base.level4.tree1.tree1.bn1.running_var. Drop parameter base.level4.tree1.tree1.bn1.num_batches_tracked. Drop parameter base.level4.tree1.tree1.conv2.weight. Drop parameter base.level4.tree1.tree1.bn2.weight. Drop parameter base.level4.tree1.tree1.bn2.bias. Drop parameter base.level4.tree1.tree1.bn2.running_mean. Drop parameter base.level4.tree1.tree1.bn2.running_var. Drop parameter base.level4.tree1.tree1.bn2.num_batches_tracked. Drop parameter base.level4.tree1.tree2.conv1.weight. Drop parameter base.level4.tree1.tree2.bn1.weight. Drop parameter base.level4.tree1.tree2.bn1.bias. Drop parameter base.level4.tree1.tree2.bn1.running_mean. Drop parameter base.level4.tree1.tree2.bn1.running_var. Drop parameter base.level4.tree1.tree2.bn1.num_batches_tracked. Drop parameter base.level4.tree1.tree2.conv2.weight. Drop parameter base.level4.tree1.tree2.bn2.weight. Drop parameter base.level4.tree1.tree2.bn2.bias. Drop parameter base.level4.tree1.tree2.bn2.running_mean. Drop parameter base.level4.tree1.tree2.bn2.running_var. Drop parameter base.level4.tree1.tree2.bn2.num_batches_tracked. Drop parameter base.level4.tree1.root.conv.weight. Drop parameter base.level4.tree1.root.bn.weight. Drop parameter base.level4.tree1.root.bn.bias. Drop parameter base.level4.tree1.root.bn.running_mean. Drop parameter base.level4.tree1.root.bn.running_var. Drop parameter base.level4.tree1.root.bn.num_batches_tracked. Drop parameter base.level4.tree1.project.0.weight. Drop parameter base.level4.tree1.project.1.weight. Drop parameter base.level4.tree1.project.1.bias. Drop parameter base.level4.tree1.project.1.running_mean. Drop parameter base.level4.tree1.project.1.running_var. Drop parameter base.level4.tree1.project.1.num_batches_tracked. Drop parameter base.level4.tree2.tree1.conv1.weight. Drop parameter base.level4.tree2.tree1.bn1.weight. Drop parameter base.level4.tree2.tree1.bn1.bias. Drop parameter base.level4.tree2.tree1.bn1.running_mean. Drop parameter base.level4.tree2.tree1.bn1.running_var. Drop parameter base.level4.tree2.tree1.bn1.num_batches_tracked. Drop parameter base.level4.tree2.tree1.conv2.weight. Drop parameter base.level4.tree2.tree1.bn2.weight. Drop parameter base.level4.tree2.tree1.bn2.bias. Drop parameter base.level4.tree2.tree1.bn2.running_mean. Drop parameter base.level4.tree2.tree1.bn2.running_var. Drop parameter base.level4.tree2.tree1.bn2.num_batches_tracked. Drop parameter base.level4.tree2.tree2.conv1.weight. Drop parameter base.level4.tree2.tree2.bn1.weight. Drop parameter base.level4.tree2.tree2.bn1.bias. Drop parameter base.level4.tree2.tree2.bn1.running_mean. Drop parameter base.level4.tree2.tree2.bn1.running_var. Drop parameter base.level4.tree2.tree2.bn1.num_batches_tracked. Drop parameter base.level4.tree2.tree2.conv2.weight. Drop parameter base.level4.tree2.tree2.bn2.weight. Drop parameter base.level4.tree2.tree2.bn2.bias. Drop parameter base.level4.tree2.tree2.bn2.running_mean. Drop parameter base.level4.tree2.tree2.bn2.running_var. Drop parameter base.level4.tree2.tree2.bn2.num_batches_tracked. Drop parameter base.level4.tree2.root.conv.weight. Drop parameter base.level4.tree2.root.bn.weight. Drop parameter base.level4.tree2.root.bn.bias. Drop parameter base.level4.tree2.root.bn.running_mean. Drop parameter base.level4.tree2.root.bn.running_var. Drop parameter base.level4.tree2.root.bn.num_batches_tracked. Drop parameter base.level4.project.0.weight. Drop parameter base.level4.project.1.weight. Drop parameter base.level4.project.1.bias. Drop parameter base.level4.project.1.running_mean. Drop parameter base.level4.project.1.running_var. Drop parameter base.level4.project.1.num_batches_tracked. Drop parameter base.level5.tree1.conv1.weight. Drop parameter base.level5.tree1.bn1.weight. Drop parameter base.level5.tree1.bn1.bias. Drop parameter base.level5.tree1.bn1.running_mean. Drop parameter base.level5.tree1.bn1.running_var. Drop parameter base.level5.tree1.bn1.num_batches_tracked. Drop parameter base.level5.tree1.conv2.weight. Drop parameter base.level5.tree1.bn2.weight. Drop parameter base.level5.tree1.bn2.bias. Drop parameter base.level5.tree1.bn2.running_mean. Drop parameter base.level5.tree1.bn2.running_var. Drop parameter base.level5.tree1.bn2.num_batches_tracked. Drop parameter base.level5.tree2.conv1.weight. Drop parameter base.level5.tree2.bn1.weight. Drop parameter base.level5.tree2.bn1.bias. Drop parameter base.level5.tree2.bn1.running_mean. Drop parameter base.level5.tree2.bn1.running_var. Drop parameter base.level5.tree2.bn1.num_batches_tracked. Drop parameter base.level5.tree2.conv2.weight. Drop parameter base.level5.tree2.bn2.weight. Drop parameter base.level5.tree2.bn2.bias. Drop parameter base.level5.tree2.bn2.running_mean. Drop parameter base.level5.tree2.bn2.running_var. Drop parameter base.level5.tree2.bn2.num_batches_tracked. Drop parameter base.level5.root.conv.weight. Drop parameter base.level5.root.bn.weight. Drop parameter base.level5.root.bn.bias. Drop parameter base.level5.root.bn.running_mean. Drop parameter base.level5.root.bn.running_var. Drop parameter base.level5.root.bn.num_batches_tracked. Drop parameter base.level5.project.0.weight. Drop parameter base.level5.project.1.weight. Drop parameter base.level5.project.1.bias. Drop parameter base.level5.project.1.running_mean. Drop parameter base.level5.project.1.running_var. Drop parameter base.level5.project.1.num_batches_tracked. Drop parameter base.fc.weight. Drop parameter base.fc.bias. Drop parameter dla_up.ida_0.proj_1.actf.0.weight. Drop parameter dla_up.ida_0.proj_1.actf.0.bias. Drop parameter dla_up.ida_0.proj_1.actf.0.running_mean. Drop parameter dla_up.ida_0.proj_1.actf.0.running_var. Drop parameter dla_up.ida_0.proj_1.actf.0.num_batches_tracked. Drop parameter dla_up.ida_0.proj_1.conv.weight. Drop parameter dla_up.ida_0.proj_1.conv.bias. Drop parameter dla_up.ida_0.proj_1.conv.conv_offset_mask.weight. Drop parameter dla_up.ida_0.proj_1.conv.conv_offset_mask.bias. Drop parameter dla_up.ida_0.up_1.weight. Drop parameter dla_up.ida_0.node_1.actf.0.weight. Drop parameter dla_up.ida_0.node_1.actf.0.bias. Drop parameter dla_up.ida_0.node_1.actf.0.running_mean. Drop parameter dla_up.ida_0.node_1.actf.0.running_var. Drop parameter dla_up.ida_0.node_1.actf.0.num_batches_tracked. Drop parameter dla_up.ida_0.node_1.conv.weight. Drop parameter dla_up.ida_0.node_1.conv.bias. Drop parameter dla_up.ida_0.node_1.conv.conv_offset_mask.weight. Drop parameter dla_up.ida_0.node_1.conv.conv_offset_mask.bias. Drop parameter dla_up.ida_1.proj_1.actf.0.weight. Drop parameter dla_up.ida_1.proj_1.actf.0.bias. Drop parameter dla_up.ida_1.proj_1.actf.0.running_mean. Drop parameter dla_up.ida_1.proj_1.actf.0.running_var. Drop parameter dla_up.ida_1.proj_1.actf.0.num_batches_tracked. Drop parameter dla_up.ida_1.proj_1.conv.weight. Drop parameter dla_up.ida_1.proj_1.conv.bias. Drop parameter dla_up.ida_1.proj_1.conv.conv_offset_mask.weight. Drop parameter dla_up.ida_1.proj_1.conv.conv_offset_mask.bias. Drop parameter dla_up.ida_1.up_1.weight. Drop parameter dla_up.ida_1.node_1.actf.0.weight. Drop parameter dla_up.ida_1.node_1.actf.0.bias. Drop parameter dla_up.ida_1.node_1.actf.0.running_mean. Drop parameter dla_up.ida_1.node_1.actf.0.running_var. Drop parameter dla_up.ida_1.node_1.actf.0.num_batches_tracked. Drop parameter dla_up.ida_1.node_1.conv.weight. Drop parameter dla_up.ida_1.node_1.conv.bias. Drop parameter dla_up.ida_1.node_1.conv.conv_offset_mask.weight. Drop parameter dla_up.ida_1.node_1.conv.conv_offset_mask.bias. Drop parameter dla_up.ida_1.proj_2.actf.0.weight. Drop parameter dla_up.ida_1.proj_2.actf.0.bias. Drop parameter dla_up.ida_1.proj_2.actf.0.running_mean. Drop parameter dla_up.ida_1.proj_2.actf.0.running_var. Drop parameter dla_up.ida_1.proj_2.actf.0.num_batches_tracked. Drop parameter dla_up.ida_1.proj_2.conv.weight. Drop parameter dla_up.ida_1.proj_2.conv.bias. Drop parameter dla_up.ida_1.proj_2.conv.conv_offset_mask.weight. Drop parameter dla_up.ida_1.proj_2.conv.conv_offset_mask.bias. Drop parameter dla_up.ida_1.up_2.weight. Drop parameter dla_up.ida_1.node_2.actf.0.weight. Drop parameter dla_up.ida_1.node_2.actf.0.bias. Drop parameter dla_up.ida_1.node_2.actf.0.running_mean. Drop parameter dla_up.ida_1.node_2.actf.0.running_var. Drop parameter dla_up.ida_1.node_2.actf.0.num_batches_tracked. Drop parameter dla_up.ida_1.node_2.conv.weight. Drop parameter dla_up.ida_1.node_2.conv.bias. Drop parameter dla_up.ida_1.node_2.conv.conv_offset_mask.weight. Drop parameter dla_up.ida_1.node_2.conv.conv_offset_mask.bias. Drop parameter dla_up.ida_2.proj_1.actf.0.weight. Drop parameter dla_up.ida_2.proj_1.actf.0.bias. Drop parameter dla_up.ida_2.proj_1.actf.0.running_mean. Drop parameter dla_up.ida_2.proj_1.actf.0.running_var. Drop parameter dla_up.ida_2.proj_1.actf.0.num_batches_tracked. Drop parameter dla_up.ida_2.proj_1.conv.weight. Drop parameter dla_up.ida_2.proj_1.conv.bias. Drop parameter dla_up.ida_2.proj_1.conv.conv_offset_mask.weight. Drop parameter dla_up.ida_2.proj_1.conv.conv_offset_mask.bias. Drop parameter dla_up.ida_2.up_1.weight. Drop parameter dla_up.ida_2.node_1.actf.0.weight. Drop parameter dla_up.ida_2.node_1.actf.0.bias. Drop parameter dla_up.ida_2.node_1.actf.0.running_mean. Drop parameter dla_up.ida_2.node_1.actf.0.running_var. Drop parameter dla_up.ida_2.node_1.actf.0.num_batches_tracked. Drop parameter dla_up.ida_2.node_1.conv.weight. Drop parameter dla_up.ida_2.node_1.conv.bias. Drop parameter dla_up.ida_2.node_1.conv.conv_offset_mask.weight. Drop parameter dla_up.ida_2.node_1.conv.conv_offset_mask.bias. Drop parameter dla_up.ida_2.proj_2.actf.0.weight. Drop parameter dla_up.ida_2.proj_2.actf.0.bias. Drop parameter dla_up.ida_2.proj_2.actf.0.running_mean. Drop parameter dla_up.ida_2.proj_2.actf.0.running_var. Drop parameter dla_up.ida_2.proj_2.actf.0.num_batches_tracked. Drop parameter dla_up.ida_2.proj_2.conv.weight. Drop parameter dla_up.ida_2.proj_2.conv.bias. Drop parameter dla_up.ida_2.proj_2.conv.conv_offset_mask.weight. Drop parameter dla_up.ida_2.proj_2.conv.conv_offset_mask.bias. Drop parameter dla_up.ida_2.up_2.weight. Drop parameter dla_up.ida_2.node_2.actf.0.weight. Drop parameter dla_up.ida_2.node_2.actf.0.bias. Drop parameter dla_up.ida_2.node_2.actf.0.running_mean. Drop parameter dla_up.ida_2.node_2.actf.0.running_var. Drop parameter dla_up.ida_2.node_2.actf.0.num_batches_tracked. Drop parameter dla_up.ida_2.node_2.conv.weight. Drop parameter dla_up.ida_2.node_2.conv.bias. Drop parameter dla_up.ida_2.node_2.conv.conv_offset_mask.weight. Drop parameter dla_up.ida_2.node_2.conv.conv_offset_mask.bias. Drop parameter dla_up.ida_2.proj_3.actf.0.weight. Drop parameter dla_up.ida_2.proj_3.actf.0.bias. Drop parameter dla_up.ida_2.proj_3.actf.0.running_mean. Drop parameter dla_up.ida_2.proj_3.actf.0.running_var. Drop parameter dla_up.ida_2.proj_3.actf.0.num_batches_tracked. Drop parameter dla_up.ida_2.proj_3.conv.weight. Drop parameter dla_up.ida_2.proj_3.conv.bias. Drop parameter dla_up.ida_2.proj_3.conv.conv_offset_mask.weight. Drop parameter dla_up.ida_2.proj_3.conv.conv_offset_mask.bias. Drop parameter dla_up.ida_2.up_3.weight. Drop parameter dla_up.ida_2.node_3.actf.0.weight. Drop parameter dla_up.ida_2.node_3.actf.0.bias. Drop parameter dla_up.ida_2.node_3.actf.0.running_mean. Drop parameter dla_up.ida_2.node_3.actf.0.running_var. Drop parameter dla_up.ida_2.node_3.actf.0.num_batches_tracked. Drop parameter dla_up.ida_2.node_3.conv.weight. Drop parameter dla_up.ida_2.node_3.conv.bias. Drop parameter dla_up.ida_2.node_3.conv.conv_offset_mask.weight. Drop parameter dla_up.ida_2.node_3.conv.conv_offset_mask.bias. Drop parameter ida_up.proj_1.actf.0.weight. Drop parameter ida_up.proj_1.actf.0.bias. Drop parameter ida_up.proj_1.actf.0.running_mean. Drop parameter ida_up.proj_1.actf.0.running_var. Drop parameter ida_up.proj_1.actf.0.num_batches_tracked. Drop parameter ida_up.proj_1.conv.weight. Drop parameter ida_up.proj_1.conv.bias. Drop parameter ida_up.proj_1.conv.conv_offset_mask.weight. Drop parameter ida_up.proj_1.conv.conv_offset_mask.bias. Drop parameter ida_up.up_1.weight. Drop parameter ida_up.node_1.actf.0.weight. Drop parameter ida_up.node_1.actf.0.bias. Drop parameter ida_up.node_1.actf.0.running_mean. Drop parameter ida_up.node_1.actf.0.running_var. Drop parameter ida_up.node_1.actf.0.num_batches_tracked. Drop parameter ida_up.node_1.conv.weight. Drop parameter ida_up.node_1.conv.bias. Drop parameter ida_up.node_1.conv.conv_offset_mask.weight. Drop parameter ida_up.node_1.conv.conv_offset_mask.bias. Drop parameter ida_up.proj_2.actf.0.weight. Drop parameter ida_up.proj_2.actf.0.bias. Drop parameter ida_up.proj_2.actf.0.running_mean. Drop parameter ida_up.proj_2.actf.0.running_var. Drop parameter ida_up.proj_2.actf.0.num_batches_tracked. Drop parameter ida_up.proj_2.conv.weight. Drop parameter ida_up.proj_2.conv.bias. Drop parameter ida_up.proj_2.conv.conv_offset_mask.weight. Drop parameter ida_up.proj_2.conv.conv_offset_mask.bias. Drop parameter ida_up.up_2.weight. Drop parameter ida_up.node_2.actf.0.weight. Drop parameter ida_up.node_2.actf.0.bias. Drop parameter ida_up.node_2.actf.0.running_mean. Drop parameter ida_up.node_2.actf.0.running_var. Drop parameter ida_up.node_2.actf.0.num_batches_tracked. Drop parameter ida_up.node_2.conv.weight. Drop parameter ida_up.node_2.conv.bias. Drop parameter ida_up.node_2.conv.conv_offset_mask.weight. Drop parameter ida_up.node_2.conv.conv_offset_mask.bias. Skip loading parameter hm.0.weight, required shapetorch.Size([64, 256, 3, 3]), loaded shapetorch.Size([256, 64, 3, 3]). Skip loading parameter hm.0.bias, required shapetorch.Size([64]), loaded shapetorch.Size([256]). Skip loading parameter hm.2.weight, required shapetorch.Size([64, 64, 3, 3]), loaded shapetorch.Size([2, 256, 1, 1]). Skip loading parameter hm.2.bias, required shapetorch.Size([64]), loaded shapetorch.Size([2]). Skip loading parameter st.0.weight, required shapetorch.Size([64, 256, 3, 3]), loaded shapetorch.Size([256, 64, 3, 3]). Skip loading parameter st.0.bias, required shapetorch.Size([64]), loaded shapetorch.Size([256]). Skip loading parameter st.2.weight, required shapetorch.Size([64, 64, 3, 3]), loaded shapetorch.Size([8, 256, 1, 1]). Skip loading parameter st.2.bias, required shapetorch.Size([64]), loaded shapetorch.Size([8]). Skip loading parameter wh.0.weight, required shapetorch.Size([64, 256, 3, 3]), loaded shapetorch.Size([256, 64, 3, 3]). Skip loading parameter wh.0.bias, required shapetorch.Size([64]), loaded shapetorch.Size([256]). Skip loading parameter wh.2.weight, required shapetorch.Size([64, 64, 3, 3]), loaded shapetorch.Size([8, 256, 1, 1]). Skip loading parameter wh.2.bias, required shapetorch.Size([64]), loaded shapetorch.Size([8]). Skip loading parameter ax.0.weight, required shapetorch.Size([64, 256, 3, 3]), loaded shapetorch.Size([256, 64, 3, 3]). Skip loading parameter ax.0.bias, required shapetorch.Size([64]), loaded shapetorch.Size([256]). Skip loading parameter ax.2.weight, required shapetorch.Size([64, 64, 3, 3]), loaded shapetorch.Size([256, 256, 1, 1]). Skip loading parameter ax.2.bias, required shapetorch.Size([64]), loaded shapetorch.Size([256]). Skip loading parameter cr.0.weight, required shapetorch.Size([64, 256, 3, 3]), loaded shapetorch.Size([256, 64, 3, 3]). Skip loading parameter cr.0.bias, required shapetorch.Size([64]), loaded shapetorch.Size([256]). Skip loading parameter cr.2.weight, required shapetorch.Size([64, 64, 3, 3]), loaded shapetorch.Size([256, 256, 1, 1]). Skip loading parameter cr.2.bias, required shapetorch.Size([64]), loaded shapetorch.Size([256]). Skip loading parameter reg.0.weight, required shapetorch.Size([64, 256, 3, 3]), loaded shapetorch.Size([256, 64, 3, 3]). Skip loading parameter reg.0.bias, required shapetorch.Size([64]), loaded shapetorch.Size([256]). Skip loading parameter reg.2.weight, required shapetorch.Size([2, 64, 1, 1]), loaded shapetorch.Size([2, 256, 1, 1]). No param conv1.weight. No param bn1.weight. No param bn1.bias. No param bn1.running_mean. No param bn1.running_var. No param bn1.num_batches_tracked. No param layer1.0.conv1.weight. No param layer1.0.conv1.bias. No param layer1.0.bn1.weight. No param layer1.0.bn1.bias. No param layer1.0.bn1.running_mean. No param layer1.0.bn1.running_var. No param layer1.0.bn1.num_batches_tracked. No param layer1.0.conv2.weight. No param layer1.0.conv2.bias. No param layer1.0.bn2.weight. No param layer1.0.bn2.bias. No param layer1.0.bn2.running_mean. No param layer1.0.bn2.running_var. No param layer1.0.bn2.num_batches_tracked. No param layer1.0.downsample.0.weight. No param layer1.0.downsample.1.weight. No param layer1.0.downsample.1.bias. No param layer1.0.downsample.1.running_mean. No param layer1.0.downsample.1.running_var. No param layer1.0.downsample.1.num_batches_tracked. No param layer1.1.conv1.weight. No param layer1.1.conv1.bias. No param layer1.1.bn1.weight. No param layer1.1.bn1.bias. No param layer1.1.bn1.running_mean. No param layer1.1.bn1.running_var. No param layer1.1.bn1.num_batches_tracked. No param layer1.1.conv2.weight. No param layer1.1.conv2.bias. No param layer1.1.bn2.weight. No param layer1.1.bn2.bias. No param layer1.1.bn2.running_mean. No param layer1.1.bn2.running_var. No param layer1.1.bn2.num_batches_tracked. No param layer2.0.conv1.weight. No param layer2.0.conv1.bias. No param layer2.0.bn1.weight. No param layer2.0.bn1.bias. No param layer2.0.bn1.running_mean. No param layer2.0.bn1.running_var. No param layer2.0.bn1.num_batches_tracked. No param layer2.0.conv2.weight. No param layer2.0.conv2.bias. No param layer2.0.bn2.weight. No param layer2.0.bn2.bias. No param layer2.0.bn2.running_mean. No param layer2.0.bn2.running_var. No param layer2.0.bn2.num_batches_tracked. No param layer2.0.downsample.0.weight. No param layer2.0.downsample.1.weight. No param layer2.0.downsample.1.bias. No param layer2.0.downsample.1.running_mean. No param layer2.0.downsample.1.running_var. No param layer2.0.downsample.1.num_batches_tracked. No param layer2.1.conv1.weight. No param layer2.1.conv1.bias. No param layer2.1.bn1.weight. No param layer2.1.bn1.bias. No param layer2.1.bn1.running_mean. No param layer2.1.bn1.running_var. No param layer2.1.bn1.num_batches_tracked. No param layer2.1.conv2.weight. No param layer2.1.conv2.bias. No param layer2.1.bn2.weight. No param layer2.1.bn2.bias. No param layer2.1.bn2.running_mean. No param layer2.1.bn2.running_var. No param layer2.1.bn2.num_batches_tracked. No param layer3.0.conv1.weight. No param layer3.0.conv1.bias. No param layer3.0.bn1.weight. No param layer3.0.bn1.bias. No param layer3.0.bn1.running_mean. No param layer3.0.bn1.running_var. No param layer3.0.bn1.num_batches_tracked. No param layer3.0.conv2.weight. No param layer3.0.conv2.bias. No param layer3.0.bn2.weight. No param layer3.0.bn2.bias. No param layer3.0.bn2.running_mean. No param layer3.0.bn2.running_var. No param layer3.0.bn2.num_batches_tracked. No param layer3.0.downsample.0.weight. No param layer3.0.downsample.1.weight. No param layer3.0.downsample.1.bias. No param layer3.0.downsample.1.running_mean. No param layer3.0.downsample.1.running_var. No param layer3.0.downsample.1.num_batches_tracked. No param layer3.1.conv1.weight. No param layer3.1.conv1.bias. No param layer3.1.bn1.weight. No param layer3.1.bn1.bias. No param layer3.1.bn1.running_mean. No param layer3.1.bn1.running_var. No param layer3.1.bn1.num_batches_tracked. No param layer3.1.conv2.weight. No param layer3.1.conv2.bias. No param layer3.1.bn2.weight. No param layer3.1.bn2.bias. No param layer3.1.bn2.running_mean. No param layer3.1.bn2.running_var. No param layer3.1.bn2.num_batches_tracked. No param layer4.0.conv1.weight. No param layer4.0.conv1.bias. No param layer4.0.bn1.weight. No param layer4.0.bn1.bias. No param layer4.0.bn1.running_mean. No param layer4.0.bn1.running_var. No param layer4.0.bn1.num_batches_tracked. No param layer4.0.conv2.weight. No param layer4.0.conv2.bias. No param layer4.0.bn2.weight. No param layer4.0.bn2.bias. No param layer4.0.bn2.running_mean. No param layer4.0.bn2.running_var. No param layer4.0.bn2.num_batches_tracked. No param layer4.0.downsample.0.weight. No param layer4.0.downsample.1.weight. No param layer4.0.downsample.1.bias. No param layer4.0.downsample.1.running_mean. No param layer4.0.downsample.1.running_var. No param layer4.0.downsample.1.num_batches_tracked. No param layer4.1.conv1.weight. No param layer4.1.conv1.bias. No param layer4.1.bn1.weight. No param layer4.1.bn1.bias. No param layer4.1.bn1.running_mean. No param layer4.1.bn1.running_var. No param layer4.1.bn1.num_batches_tracked. No param layer4.1.conv2.weight. No param layer4.1.conv2.bias. No param layer4.1.bn2.weight. No param layer4.1.bn2.bias. No param layer4.1.bn2.running_mean. No param layer4.1.bn2.running_var. No param layer4.1.bn2.num_batches_tracked. No param adaption3.weight. No param adaption2.weight. No param adaption1.weight. No param adaption0.weight. No param adaptionU1.weight. No param deconv_layers1.0.weight. No param deconv_layers1.1.weight. No param deconv_layers1.1.bias. No param deconv_layers1.1.running_mean. No param deconv_layers1.1.running_var. No param deconv_layers1.1.num_batches_tracked. No param deconv_layers2.0.weight. No param deconv_layers2.1.weight. No param deconv_layers2.1.bias. No param deconv_layers2.1.running_mean. No param deconv_layers2.1.running_var. No param deconv_layers2.1.num_batches_tracked. No param deconv_layers3.0.weight. No param deconv_layers3.1.weight. No param deconv_layers3.1.bias. No param deconv_layers3.1.running_mean. No param deconv_layers3.1.running_var. No param deconv_layers3.1.num_batches_tracked. No param deconv_layers4.0.weight. No param deconv_layers4.1.weight. No param deconv_layers4.1.bias. No param deconv_layers4.1.running_mean. No param deconv_layers4.1.running_var. No param deconv_layers4.1.num_batches_tracked. No param ax.4.weight. No param ax.4.bias. No param ax.6.weight. No param ax.6.bias. No param ax.8.weight. No param ax.8.bias. No param cr.4.weight. No param cr.4.bias. No param cr.6.weight. No param cr.6.bias. No param cr.8.weight. No param cr.8.bias. No param hm.4.weight. No param hm.4.bias. No param hm.6.weight. No param hm.6.bias. No param hm.8.weight. No param hm.8.bias. No param st.4.weight. No param st.4.bias. No param st.6.weight. No param st.6.bias. No param st.8.weight. No param st.8.bias. No param wh.4.weight. No param wh.4.bias. No param wh.6.weight. No param wh.6.bias. No param wh.8.weight. No param wh.8.bias. loaded ../dir_of_ckpt/ckpt_wireless/processor_best.pth, epoch 100 0%| | 0/2 [00:00<?, ?it/s](3000, 5) 50%|████████████████████████████████████████████▌ | 1/2 [00:00<00:00, 2.30it/s](3000, 5) 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2/2 [00:00<00:00, 2.21it/s]
查看保存的文件,发现还是原始图片,表格结构完全没有识别。
另外,请问是哪里的问题?预训练模型从链接中下载。