search

MingtaoFu / gliding_vertex

The implementation of paper "Gliding vertex on the horizontal bounding box for multi-oriented object detection".
271 stars 63 forks source link

您好，我想要训练ICDAR2015数据，我用您给的方法变换数据集不太对 #7

Closed shenhongjuan closed 4 years ago

MingtaoFu commented 4 years ago

请提供具体的错误信息。

shenhongjuan commented 4 years ago

Traceback (most recent call last): File "pp.py", line 3, in img_dic = collect_unaug_dataset(os.path.join("/home/dc2-user/shj/gliding_vertex-master/gliding_vertex-master/Challenge4/", "ch4_training_localization_transcription_gt/")) File "/home/dc2-user/shj/gliding_vertex-master/gliding_vertex-master/txt2json.py", line 143, in collect_unaug_dataset img_dic[key].append( txt[:-4] ) KeyError: '###\n'

MingtaoFu commented 4 years ago

您好，请阅读README，txt的格式为 x1 y1 x2 y2 x3 y3 x4 y4 cls diff。这个问题应该是由于该数据集只有一个类，因此没有cls导致的。另外，这个脚本的diff是0/1的，而ICDAR的标注如果是diff则是###，也应该进行修改。