Open daili650 opened 2 years ago
能留一个联系方式吗
KeyError: 9 是不是你类别数量和对应的label_dict.py没改
KeyError: 9 是不是你类别数量和对应的label_dict.py没改
用的是自己制作的同一个数据集
能留一个联系方式吗
1466908765@qq.com,你把联系方式发我,我加你吧
所以你需要改类别总数,已经修改对应的label_dict.py
所以你需要改类别总数,已经修改对应的label_dict.py
我已经修改过了,成功训练过一个网络,第二次才会出现问题
你可以debug,看它找的是label_dict.py里的哪个数据集
你可以debug,看它找的是label_dict.py里的哪个数据集
我是模仿的dota数据集,修改了label_dict.py的DOTA数据集,
但是在test_dota_base.py中,self.cfgs.CLASS_NUM依旧为15,
Traceback (most recent call last): File "", line 1, in
File "D:\PyCharm 2020.1\plugins\python\helpers\pydev_pydev_bundle\pydev_umd.py", line 197, in runfile
pydev_imports.execfile(filename, global_vars, local_vars) # execute the script
File "D:\PyCharm 2020.1\plugins\python\helpers\pydev_pydev_imps_pydev_execfile.py", line 18, in execfile
exec(compile(contents+"\n", file, 'exec'), glob, loc)
File "I:/projects/RotationDetection-new/tools/gwd/test_dota.py", line 33, in
tester.eval()
File "I:/projects/RotationDetection-new/tools/gwd/test_dota.py", line 25, in eval
self.test_dota(det_net=gwd, real_test_img_list=real_test_img_list, txt_name=txt_name)
File "I:\projects\RotationDetection-new\tools\test_dota_base.py", line 310, in test_dota
proc = self.test(int(gpu_id), split_records, det_net, result_queue)
File "I:\projects\RotationDetection-new\tools\test_dota_base.py", line 281, in test
float(threshold[self.label_name_map[sub_class]]), 0)
KeyError: 9
train一个网络后,能成功test,接着训练其他网络,能成功训练,但是不能测试,同时前面的网络也不能测试了