图片预处理时出现错误

[yx618]

我在处理dota1.5数据集的时侯，出现了如下错误，请问一下我该如何解决？谢谢

File "/home/speech/cipan/s2anet/DOTA_devkit/convert_dota_to_mmdet.py", line 66, in convert_dota_to_mmdet
bboxes, labels, bboxes_ignore, labels_ignore = parse_ann_info(label_path, img_name)
File "/home/speech/cipan/s2anet/DOTA_devkit/convert_dota_to_mmdet.py", line 32, in parse_ann_info
labels.append(label_ids[class_name])
KeyError: 'container-crane'

[csuhan]

添加 'container-crane'：

s2anet/DOTA_devkit/convert_dota_to_mmdet.py

Lines 10 to 14 in d287096

	wordname_15 = ['plane', 'baseball-diamond', 'bridge', 'ground-track-field',
	'small-vehicle', 'large-vehicle', 'ship', 'tennis-court',
	'basketball-court', 'storage-tank', 'soccer-ball-field', 'roundabout',
	'harbor', 'swimming-pool', 'helicopter']