使用Tensorflow对象检测API训练自定义模型的问题

时间:2020-09-04 00:41:27

标签: tensorflow object-detection object-detection-api loss tensorflow-model-garden

我想训练只能检测人的自定义模型。
损失没有收敛或收敛到比我预期的高。
这是我的环境。

Tensorflow version : 1.13.1
Pre-trained model : SSD mobilenet v2 quantized 300x300 coco
Learning rate optimizer : Adam optimizer / exponential decay learning rate

我已经尝试过这些东西。

Add more images
Add gaussian noise for 1/3 of data set
Reduce loss 0.004 to 0.000001 by stages.
Change learning rate optimizer rms prop to adam
change tensorflow version 1.15.0 to 1.13.1

但是那些东西并不能解决我的问题。
顺便说一句,我发现了一些奇怪的东西。
我的数据集中的某些图像有问题。
我准备的所有图像都来自可可数据集。
例如,this image看起来没有人。
我将注释文件.json(从coco)直接转换为.xml,所以我没有意识到。
终于在图片中找到了公交车司机,他真的很难认出。
这样的图像会引起问题吗?
如果没有,该如何解决此问题?

0 个答案:

没有答案