报错信息:
Traceback (most recent call last):
File "/home/PycharmProjects/yolov4/3-L_m-lite-Att-later_marker2400/get_dr_txt.py", line 119, in <module>
yolo = mAP_Yolo()
File "/home/PycharmProjects/yolov4/3-L_m-lite-Att-later_marker2400/yolo.py", line 56, in __init__
self.generate()
File "/home/PycharmProjects/yolov4/3-L_m-lite-Att-later_marker2400/yolo.py", line 93, in generate
self.net.load_state_dict(state_dict)
File "/home/.conda/envs/ultralytics_yolov3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1045, in load_state_dict
self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for YoloBody:
Missing key(s) in state_dict: "feat0_att.fc.0.weight", "feat0_att.fc.2.weight", "feat1_att.fc.0.weight", "feat1_att.fc.2.weight", "feat2_att.fc.0.weight", "feat2_att.fc.2.weight", "upsample1_att.fc.0.weight", "upsample1_att.fc.2.weight", "upsample2_att.fc.0.weight", "upsample2_att.fc.2.weight".
Unexpected key(s) in state_dict: "feat0_att.convs.0.0.weight", "feat0_att.convs.0.0.bias", "feat0_att.convs.0.1.weight", "feat0_att.convs.0.1.bias", "feat0_att.convs.0.1.running_mean", "feat0_att.convs.0.1.running_var", "feat0_att.convs.0.1.num_batches_tracked", "feat0_att.convs.1.0.weight", "feat0_att.convs.1.0.bias", "feat0_att.convs.1.1.weight", "feat0_att.convs.1.1.bias", "fe