WuLiXiaoKeAi的博客

原创 csv2coco.py报错 annotation[‘category_id‘] =int(classname_to_id[label])KeyError: 5

原因：pandas读取csv时，遇到数字字符会将其转化为int型:按照classname_to_id字典去索引就不行了，因为label的数据格式变成了int 解决办法： 将main函数中 annotations=pd.read_csv(csv_file,header=None).values 改为 ...