RotationDetection
RotationDetection copied to clipboard
train多个网络后,test出现问题
Traceback (most recent call last):
File "", line 1, in
能留一个联系方式吗
KeyError: 9 是不是你类别数量和对应的label_dict.py没改
KeyError: 9 是不是你类别数量和对应的label_dict.py没改
用的是自己制作的同一个数据集
所以你需要改类别总数,已经修改对应的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,