首页 文章

管理tensorflow对象检测API中的类

提问于
浏览
1

我正在开展一个项目,需要识别视频中的人或来自摄像机的直播流 . 我目前正在使用具有python的tensorflow对象识别API,并且我尝试了不同的预训练模型和冻结推理图 . 我只想识别人和汽车,所以我不需要我的神经网络来识别基于mobilenet或rcnn的冷冻推理图所带来的所有90个类,因为这似乎会减慢这个过程,这90个中有89个我的项目中不需要课程 . 我是否必须训练自己的模型,或者有没有办法修改推理图和现有模型?对于你们中的一些人来说,这可能是一个noob问题,但请注意,我已经使用了tensorflow和机器学习一个月 . 提前致谢

1 回答

  • 0

    缩小最后一层以输出1或2个类不太可能产生大的加速 . 这是因为大多数计算都在中间层 . 您可以缩小中间层,但这会导致精度较差 .

相关问题