前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >目标检测框架:支持TF2的TF Object Detection

目标检测框架:支持TF2的TF Object Detection

作者头像
小白学视觉
发布2022-09-28 11:01:48
4500
发布2022-09-28 11:01:48
举报

一直以来,大家最常用的目标检测库是港中文的mmdetectionFacebook的detectron2库,不过这两个库都是基于PyTorch的,PyTorch的模型部署方面还是稍差于TensorFlow,如果想用TensorFlow的目标检测模型,最好的还是选择还是谷歌官方的TF Object Detection库:

目前随着TensorFlow 2x的到来,TF Object Detection库也支持TF2了,而且最重要的是还兼容TensorFlow 1.x,这真是非常nice。不过官方还是建议大家使用最新的TF2来训练模型,主要原因如下:

  • 最新的模型只会在TF2中更新,未来还会继续更新;
  • TF2训练得到的模型效果和TF1几乎没有差别;
  • TF2可以更容易地使用GPU和TPU进行分布式训练;
  • TF2的Eager模式使debug更容易;

除了支持TF2外,TF2也新增了更多的模型如CenterNet和EfficientDet,具体Model Zoo如下所示:

不过,如果你想使用TF1,那么你可以查看对应的Model Zoo(https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/tf1_detection_zoo.md),要比TF2支持的模型少一些。

另外,贴心的开发者还给了详细的入门教程Colab:

  • 训练:在自己的数据集上fine-tune一个检测器;
  • 推理:用model zoo中的模型进行推理预测;
  • 移动端部署:Fine-tune一个检测器,并用TensorFlow Lite部署

更多见https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/tf2.md

本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2022-06-18,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 小白学视觉 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档