将分类模型用于对象检测框架

时间:2018-07-07 17:58:28

标签: python tensorflow object-detection

我目前正在尝试使用Google Tensorflow中的Object Detection framework。 我有一个类似MNIST的经过训练的模型。将其用作分类检查点的最简单方法是什么? 据我目前的了解,我可以使用分类或对象检测检查点。 只是不知道如何使用分类检查点,因为我认为网络的结构是由

中的类型指定的
 feature_extractor {
  type: "ssd_mobilenet_v2"
  ...

我必须使用this提供自己的模型类型吗?

所有预训练模型都有很多层,而像MNIST这样的数据集只有3个,非常容易训练。

通常的目标是在带有边框的白色背景上检测数学符号。分类部分很容易,但是尝试通过对象检测对其进行扩展似乎很困难。使用在真实世界的图像上训练的用于对象检测的预训练模型似乎比刮擦要好,但总体而言还很糟糕。

任何想法表示赞赏!

0 个答案:

没有答案