object_detection“使用教程”翻译(二):Tensorflow检测模型zoo

我们提供了在COCO数据集上预训练好的检测模型的集合【注:针对COCO数据集,已经开源了好几个检测模型,具体可以到项目中查看】。如果您对COCO数据集中已经存在的类别(例如,人类,汽车等)感兴趣,那么这些模型可以开箱即用。在训练新数据集时,它们也可用于初始化模型。 在下表中,我们列出了这样的预先训练好的模型,包括:

  • 与samples / configs目录中的配置文件的文件名相对应的模型
  • 一个包含预先训练好的模型的tar.gz文件的下载链接
  • 模型的速度({slow,medium,fast}之一)
  • 检测器的性能——COCO数据测量的mAP值。在这里,越高越好,我们只报告四舍五入到最接近的整数的bounding boxes的mAP值
  • 输出类型(目前只有Boxes)
    【注:链接在本文末尾】
    这里写图片描述

您可以通过以下方式解压每个tar.gz文件

tar -xzvf ssd_mobilenet_v1_coco.tar.gz

在解压后的目录里,你会发现:

  • 一个graph proto (graph.pbtxt)
  • 一个checkpoint (model.ckpt.data-00000-of-00001,model.ckpt.index, model.ckpt.meta)
  • 一个各种权重已经作为常量baked到图中的冻结了的graph proto (frozen_inference_graph.pb) 用于开箱既用 inference (在Jupyter notebook试一试!)

[注:以下是各个模型的下载链接]:
1. ssd_mobilenet_v1_coco
2. ssd_inception_v2_coco
3. rfcn_resnet101_coco
4. faster_rcnn_resnet101_coco
5. faster_rcnn_inception_resnet_v2_atrous_coco
6. faster_rcnn_nas_coco

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值