weixin_44226279的博客

原创 解决labelme_json_to_dataset批量多类别转化遇到的问题

使用 labelme 批量标注时会遇到一个问题：labelme 的 json 只保存一张图中存在的标注，这意味着一张图中不包含所有类别的时候，它的标注颜色（值）不一样，相当于类别标注错误了，这样的数据肯定是不能送去训练的。（比如这批数据总共有类别 {‘1’,‘2’,‘3’,‘4’,‘5’},而这张图只包含{‘1’,‘3’,‘5’},json中只保存135,在转化成mask图时图中类别和对应的值会变成{ ‘1’->1 , ‘3’->2 , ‘5’->3 }，这样类别就错误了）解决方法