-
Notifications
You must be signed in to change notification settings - Fork 9
Description
测试DOTA1.0数据集时,加载checkpoint “strip_rcnn_s_dota”会报:
unexpected key in source state_dict: roi_head.bbox_head.reg_xy_wh_convs.1.conv_spatial1.weight, roi_head.bbox_head.reg_xy_wh_convs.1.conv_spatial1.bias, roi_head.bbox_head.reg_xy_wh_convs.1.conv_spatial2.weight, roi_head.bbox_head.reg_xy_wh_convs.1.conv_spatial2.bias
missing keys in source state_dict:
roi_head.bbox_head.reg_xy_wh_convs.1.strip_conv1.weight, roi_head.bbox_head.reg_xy_wh_convs.1.strip_conv1.bias, roi_head.bbox_head.reg_xy_wh_convs.1.strip_conv2.weight, roi_head.bbox_head.reg_xy_wh_convs.1.strip_conv2.bias, roi_head.bbox_head.reg_theta_fcs.0.weight, roi_head.bbox_head.reg_theta_fcs.0.bias, roi_head.bbox_head.reg_theta_fcs.1.weight, roi_head.bbox_head.reg_theta_fcs.1.bias
似乎缺少了 theta 的两个 fc 层,以及conv_spatial1/conv_spatial2不是StripHead 里的 StripBlock