搭建自己的物体检测模型系列(1)环境的搭建和测试

本文详细介绍如何使用TensorFlow Object Detection API训练自定义物体检测模型。以Windows 10 64位系统为例,配备Nvidia GTX 1080 Ti显卡,采用TensorFlow 1.10.0版本,通过实际操作步骤解决配置过程中遇到的问题。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

说明

本系列主要使用谷歌开源的 Tensoflow Object Detection API来进行,使用自己的数据集来训练自己的模型,从而完成目标物体检测的任务!
主要参考博主dy_guox的一下几篇博客视频讲解以及常见问题汇总,由于环境的不同,过程中也遇到了许许多多的问题,因此来记录下自己的配置过程,方便下次使用!(非常感谢上面这位博主。)
https://blog.youkuaiyun.com/dy_guox/article/details/79081499
https://blog.youkuaiyun.com/dy_guox/article/details/79111949
https://blog.youkuaiyun.com/dy_guox/article/details/80139981
https://blog.youkuaiyun.com/dy_guox/article/details/80192343

1、环境说明

操作系统:Windows10 64位
内存:16G
显卡:Nvidia GTX1080 Ti
Tensorflow:1.10.0

这里的环境配置和这个文章是一模一样的,仅仅是Tensorflow的版本不一样,博客里是1.8.0.(Tensorflow版本最低是1.10.0,要不然在使用export_inference_graph.py函数导出模型的时候会报错)。

2、下载Tensorflow Object Detection API

下载地址在Github上:https://github.com/tensorflow/models, 从github上下载项目(右上角“Clone or download”-“DownloadZIP”),下载到本地目录(避免中文),解压并修改文件名为models。

下载慢的可以去百度网盘去下载
链接:https://pan.baidu.com/s/1lVb1O0s-MVgH1mmwjgmzJw
提取码:bs03

对于训练模型的下载,下载慢的也可以去百度网盘下载
链接:https://pan.baidu.com/s/1x5Rm7S9RmdZMtgn8UDFZog
提取码:j4nz

:这里的模型仅仅是ssd_mobilenet_v1_coco_2017_11_17模型,对于下载好的模型来进行测试,需要把下载好的模型放到models\research\object_detection路径的下面并且需要注释掉object_detection_tutorial.ipynb文件中的下面两行即可
opener = urllib.request.URLopener()
opener.retrieve(DOWNLOAD_BASE + MODEL_FILE, MODEL_FILE)

3、Protobuf 安装与配置

下面的所有配置参考这篇文章:https://blog.youkuaiyun.com/qq_40196164/article/details/94593703

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值