训练自己的数据集，出来的模型，效果特别差

[zhouzhubin]

你好，我也是按照你写的步骤训练自己的数据集，有17个类别的，训练图片张数达到25000多张，bath_size 用24，训练了有110000多步，没有用fine_tune，但是训练出来的模型，效果很差，得分很低，而且类别标注也是不对的。另外我最开始用的是fine_tune的，效果和这一样，所以想问一下你，我最可能在哪个地方出问题了。麻烦你了

[zhouzhubin]

@XiangGuo1992 麻烦了啊

[zhouzhubin]

@XiangGuo1992 我训练的loss一直在1到2之间徘徊

[zhouzhubin]

@XiangGuo1992 相同的数据集，我用github上的tensorflow-SSD进行训练，bath_size 也用的是24，只训练了13000多次，效果就挺不错的了，有90%多以上了。但是用官方的API，训练了11万多次，效果会那么差，很郁闷！

[XiangGuo1992]

这个我也不清楚。如果步骤都正确的话，不会差别这么大的。你可以试试去官方的API下面提问，应该能得到更专业的解释

[zhouzhubin]

@XiangGuo1992 我这个数据集也不是标准的voc数据集格式，但是有效的数据信息都有的

[XiangGuo1992]

你确认有效信息都一致的话，那就有点蹊跷了，有没有试过API里面别的model？或者有可能是选择的 pretrained model 的原因。API里下载的是基于特定数据集的预训练模型，跟你做的tensorflow-SSD训练起始条件不一样。这是我的猜测，仅供参考。

[zhouzhubin]

@XiangGuo1992 我两种都试过，我是看了你的博客，然后我去掉了预训练模型，训练了十一万多次，结果跟加载预训练模型的效果差不多，都比较差

[zhouzhubin]

[zhouzhubin]

@XiangGuo1992 你试过多类别训练吗？是不是因为我的类别数比较多？但是也不对voc有21类，这个我都搞了两周，tensflow版本也是从1.5升到1.7了

[XiangGuo1992]

我目前最多是4类。建议去API下面反映一下这个问题

[zhouzhubin]

@XiangGuo1992 嗯嗯，好的，谢谢了。我想问一下，那个tfrecord里，只要这些数据就可以了吗？

[zhouzhubin]

@XiangGuo1992 你好，我不知道是不是官方代码里有bug，我用了ssdlite_mobilenet_v2进行训练，相同的步骤，训练了一万多次，就出了效果了

[yff1995]

请问楼主现在用tensorflow最高什么版本的可以正常使用object_detection API呢？我之前试过1.3 1.4都存在很多问题，现在换到cuda10.0+cudnn7.4.1+tensorflow1.13了可以正常完成训练，但是在测试时使用楼主给的代码都会出现一些问题，不管是用jupyter notebook或是spyter或者pycharm ,个人感觉是tensorflow版本问题，跪求楼主指点一下。万分感谢！！！