ssd-tensorflow报错 labels.append(int(VOC_LABELS[label][0])) KeyError:‘liqin’ 问题解决

修复PascalVOC到TFRecords转换问题

原创于 2019-12-23 16:17:23 发布 · 1.1k 阅读

CC 4.0 BY-SA版权

4 篇文章

订阅专栏

本文介绍了一种解决在将PascalVOC数据集转换为TensorFlow记录格式过程中遇到的问题的方法。具体是在pascalvoc_to_tfrecords.py文件的第103行，通过将标签直接设置为1来解决类别编号问题。

部署运行你感兴趣的模型镜像

参考文章:https://blog.youkuaiyun.com/qigeyonghuming_1/article/details/89205690
找到datasets文件夹下的pascalvoc_to_tfrecords.py文件。
找到103行

labels.append(1)#int(VOC_LABELS[label][0])label对应的类别编号，此处直接使用1.没什么特殊意义

解决！

您可能感兴趣的与本文相关的镜像

TensorFlow-v2.15

TensorFlow

TensorFlow 是由Google Brain 团队开发的开源机器学习框架,广泛应用于深度学习研究和生产环境。它提供了一个灵活的平台,用于构建和训练各种机器学习模型

1 条评论