哪种Tensorflow物体检测模型最适合卫星图像?

时间:2017-12-09 11:54:25

标签: tensorflow object-detection

我有一个有卫星图像的数据集。因此,数据集与用于对象检测的常用图像数据集完全不同。我训练了ssd_mobilenet_v1_pets模型,但训练有素的模型表现非常糟糕。

这是否意味着ssd_mobilenet_v1_pets不适合卫星图像?在我的情况下,TF对象检测中的哪一个其他可用模型会更好?

1 个答案:

答案 0 :(得分:1)

模型的最佳选择取决于您的确切要求和部署环境。 SSD Mobilenet(以及其他SSD型号)可以非常快速地执行推理,但精度较低。它们非常适合需要快速/实时推理的情况或计算能力有限的情况(即移动电话或物联网)。相比之下,更快的RCNN或RFCN模型将产生更准确的结果,但运行速度较慢。

考虑尝试更快的RCNN Resnet 101模型。如果您需要更多详细信息,请查看有关对象检测架构的速度准确性权衡的model zoothis paper