Open gottastudy opened 4 months ago
使用链接提供NUDT的.tar和IRSTD.pkl不能使用,提示没有 File "\MSHNet\main.py", line 123, in init self.model.load_state_dict(weight['state_dict'])
~~^^^^^^^^^^^^^^ KeyError: 'state_dict' 训练得到的NUDT-SIRST结果为 2024-06-15-11-11-50 - 0000 - IoU 0.0017 - PD 0.3778 - FA 254317.8007 2024-06-15-11-12-55 - 0001 - IoU 0.0538 - PD 0.3095 - FA 4244.0116 2024-06-15-11-13-59 - 0002 - IoU 0.0662 - PD 0.2843 - FA 5671.4207 2024-06-15-11-14-54 - 0003 - IoU 0.1980 - PD 0.2249 - FA 1346.7260 2024-06-15-11-15-48 - 0004 - IoU 0.2395 - PD 0.1862 - FA 1204.0655 2024-06-15-11-16-42 - 0005 - IoU 0.3870 - PD 0.1451 - FA 407.4143 2024-06-15-11-17-50 - 0006 - IoU 0.4114 - PD 0.1008 - FA 40.6518 2024-06-15-11-18-58 - 0007 - IoU 0.4695 - PD 0.1106 - FA 194.5725 2024-06-15-11-20-04 - 0008 - IoU 0.4854 - PD 0.0990 - FA 154.7250 2024-06-15-11-21-21 - 0009 - IoU 0.5054 - PD 0.0877 - FA 151.9904 2024-06-15-11-24-38 - 0012 - IoU 0.5274 - PD 0.0703 - FA 174.4420 2024-06-15-11-28-46 - 0016 - IoU 0.5404 - PD 0.0542 - FA 199.1456 2024-06-15-11-29-49 - 0017 - IoU 0.5528 - PD 0.0513 - FA 161.4812 2024-06-15-11-32-05 - 0019 - IoU 0.5689 - PD 0.0465 - FA 164.1469 2024-06-15-11-35-35 - 0022 - IoU 0.5736 - PD 0.0398 - FA 149.3707 2024-06-15-11-36-43 - 0023 - IoU 0.6054 - PD 0.0380 - FA 74.9611 2024-06-15-11-37-51 - 0024 - IoU 0.6092 - PD 0.0350 - FA 45.7764 2024-06-15-11-39-05 - 0025 - IoU 0.6126 - PD 0.0343 - FA 64.6201 2024-06-15-11-42-31 - 0028 - IoU 0.6339 - PD 0.0307 - FA 43.4784 2024-06-15-11-55-04 - 0039 - IoU 0.6430 - PD 0.0227 - FA 57.9788 2024-06-15-11-59-45 - 0043 - IoU 0.6555 - PD 0.0200 - FA 26.3122 2024-06-15-12-05-26 - 0048 - IoU 0.6635 - PD 0.0187 - FA 55.8876 2024-06-15-12-10-11 - 0052 - IoU 0.6656 - PD 0.0174 - FA 48.1893 2024-06-15-12-11-13 - 0053 - IoU 0.6787 - PD 0.0174 - FA 51.0618 2024-06-15-12-26-07 - 0066 - IoU 0.6801 - PD 0.0135 - FA 36.6533 2024-06-15-12-29-44 - 0069 - IoU 0.6803 - PD 0.0131 - FA 45.0180 2024-06-15-12-32-01 - 0071 - IoU 0.6870 - PD 0.0125 - FA 29.3456 2024-06-15-12-51-34 - 0088 - IoU 0.6904 - PD 0.0102 - FA 32.0572 2024-06-15-12-53-51 - 0090 - IoU 0.6915 - PD 0.0103 - FA 51.7282 2024-06-15-12-55-08 - 0091 - IoU 0.6977 - PD 0.0102 - FA 52.2338 2024-06-15-12-58-30 - 0094 - IoU 0.7048 - PD 0.0098 - FA 41.2033 2024-06-15-13-11-54 - 0106 - IoU 0.7154 - PD 0.0088 - FA 36.6533 2024-06-15-13-18-45 - 0112 - IoU 0.7166 - PD 0.0081 - FA 20.0616 2024-06-15-13-25-42 - 0118 - IoU 0.7198 - PD 0.0079 - FA 41.8697 2024-06-15-13-38-16 - 0129 - IoU 0.7217 - PD 0.0072 - FA 30.4257 2024-06-15-13-47-17 - 0137 - IoU 0.7266 - PD 0.0067 - FA 25.1632 2024-06-15-13-48-23 - 0138 - IoU 0.7317 - PD 0.0069 - FA 37.0209 2024-06-15-13-49-40 - 0139 - IoU 0.7375 - PD 0.0068 - FA 28.3804 2024-06-15-13-55-09 - 0144 - IoU 0.7416 - PD 0.0065 - FA 18.7518 2024-06-15-13-59-44 - 0148 - IoU 0.7418 - PD 0.0064 - FA 31.9194 2024-06-15-14-27-50 - 0173 - IoU 0.7435 - PD 0.0054 - FA 22.3367 2024-06-15-14-45-19 - 0189 - IoU 0.7543 - PD 0.0050 - FA 26.4731 2024-06-15-16-06-14 - 0247 - IoU 0.7561 - PD 0.0038 - FA 22.2907 2024-06-15-16-20-51 - 0257 - IoU 0.7581 - PD 0.0037 - FA 24.5198 2024-06-15-16-22-18 - 0258 - IoU 0.7655 - PD 0.0037 - FA 23.0261 2024-06-15-18-02-07 - 0331 - IoU 0.7657 - PD 0.0028 - FA 20.3144 2024-06-15-18-06-34 - 0335 - IoU 0.7686 - PD 0.0029 - FA 22.9112 2024-06-15-18-07-40 - 0336 - IoU 0.7739 - PD 0.0029 - FA 26.2203 2024-06-15-18-28-11 - 0354 - IoU 0.7746 - PD 0.0027 - FA 16.9134
请问这个问题解决了吗,同样遇到了这个问题
解决了,把load_state_dict(weight['state_dict']),改成,self.model.load_state_dict(weight),去掉state_dict就行
使用链接提供NUDT的.tar和IRSTD.pkl不能使用,提示没有 File "\MSHNet\main.py", line 123, in init self.model.load_state_dict(weight['state_dict'])