关于object_detection的官网运行案例

本文档详细介绍了如何使用TensorFlow的object_detection API训练 Faster R-CNN ResNet101 模型。从下载代码库到安装依赖,再到配置模型参数,最后执行训练和评估任务,每个步骤都有清晰的说明,包括解决CUDA和CuDNN版本不兼容问题的方法。
摘要由CSDN通过智能技术生成

pip install tensorflow-tensorboard

首先需要把https://github.com/tensorflow/models/tree/r1.5的代码下载下来

然后进入到https://github.com/tensorflow/models/tree/r1.5/research/object_detection,下面有个README.md

1.installation:

    搭建GPU 环境,反正这个地方,搭建GPU的时候会遇到很多问题,自己多在百度上搜索和尝试,然后在终端依次输入以下命令:

pip install tensorflow-gpu

sudo apt-get install protobuf-compiler python-pil python-lxml

sudo pip install jupyter

sudo pip install matplotlib

sudo pip install pillow

sudo pip install lxml

sudo pip install jupyter

sudo pip install matplotlib

其次 # From tensorflow/models/research/,也就是在research下面的文件下输入:

protoc object_detection/protos/*.proto --python_out=.

export PYTHONPATH=$PYTHONPATH:`pwd`:`pwd`/slim
python object_detection/builders/model_builder_test.py
运行完成会出现

 

2. Configuring an object detection pipeline:

         定义obeject_detection pipeline,就是给他一个配置文件,在配置文件里面写明需要用到的哪些model,可以参考object_detection/samples/configs,可以看到configs下面有很多标准的configs文件,可以根据这些成型的文件进行相应的修改。根据自己的经验来看,再进行自己数据训练之前,最好把官网的模型走一遍,以确定框架本身没有什么问题。

以faster_rcnn_resnet101_voc07.config为例子。

里面的model,包含了模型的本身的超参数

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

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值