object detection API完整配置教程

@object detection完整配置教程

研一下学期用ubuntu16.04配置过一次tensorflow object detection 很遗憾试过很多次都失败了,不了了之,当初记得是tensorflow版本的问题。我用的1.4,cuda是8.cudnn是6,那个时候还只是照着书敲keras还很有成就感的纯小白,研二上学期开组会听师兄开组会说他写的论文(好像还是SCI)做实验就用的这个API,决定再试一次。
这次顺利的多吧,总共花了国庆节不到两天,和本科同学开黑玩lol还花了好多时间,就当劳逸结合吧哈哈哈(我卡莎玩的还行)
在demo那两张结果图运行成功的那一刻,还是有点小激动的,毕竟菜鸡还是第一次看见完整的目标检测结果图。总的来说,遇到问题就百度,基本都靠CSDN,一个不行就试试另一个,今天不行就明天,总能成功的,但是不能一根筋,要学会变通
我的按照这个一步一步来,肯定可以成功

运行环境 win10正版 显卡1050Ti 显卡驱动版本到官网上输入对应信息下载最新的
Cuda是可以同时存在多个的我电脑上就同时有10.1 ,9.0,9.2
9.2是pythorch1.2需要的 9.0是tensorflow需要的,1.12都不支持9.2
装好cuda后环境变量是自动配置的 最后实现用的是pycharm
Tensorflow-gpu版本用的是1.10
我用的笔记本 惠普暗夜精灵4 pro 跑的深度学习,师门唯一1080Ti不在我这,先凑合着用吧,欸

我下面说的主要是研二开学开始做的一些工作,因为到了研二觉得自己研一实在太水了,老师又要求12月之前发论文投会议(当然比较水),时间还是很紧张的,所以国庆节都不敢出去玩
下面说流程吧(废话有点多)
一开始我的电脑是cuda8加cudnn6,配置太低了高版本的tf,pytorch都用不了,keras新手用用对这方面有个概念就行,毕竟这两个是现在最火的两个框架,下面想用用caffe和mxnet,不过框架至少要精通一种,python语法,库很多,用到就查,我觉得不用可以花很多时间去学,不像C语言和C++,不过cuda库是用这个写的,以后从事这个方面肯定要学,之前觉得cuda是只能单独存在一个,觉得很高端,原来只是英伟达自家的库,我觉得在win下安装卸载比在ubuntu下面方面,卸载直接去控制面板下面删,cudnn直接拷贝复制到对应的文件夹里就好了,查看版本也容易 cmd 里nvcc -V就行,目前我笔记本也够用

肯定把之前装的cuda,cudnn卸载掉,然后开始最坑的安装cuda10,现在想想完全没必要卸载,怎么装都装不上,折腾了得一两天吧,我还把VS2015卸载了安装的VS2017,驱动卸载过两次,然后我双屏直接黑屏了,我把英伟达声卡也卸载了,然后插耳机也没声音只能外放,最后还是到官网上下的最新版驱动里面由系统,声卡驱动,图形驱动三个,update core也装不上,也不知道什么原因

Cuda10里面基本cuda组件都安不上去,既然10不行,算了试试9把,(我现在想想是不是我笔记本显卡太low,有机会撺掇导师买个titan x或者2080ti再试试哈哈哈哈),这里推荐一个驱动卸载程序,用得挺好的叫DDU,一定要再安全模式下,具体自行百度,还是那句话一步一步照做,安了9.2和匹配cuda,import torch没有问题

下面就到正题了,装tensorflow object detection这个api,具体流程我是参照这个来的

https://blog.csdn.net/sarsscofy/article/details/81111815
大概说下吧 前面都进行的比较顺利
下载models,protobuf(这个我到官网上下载的最新的,结果后面跟tensorflow1.10版本不兼容,卸载了下了个3.6)安装一些库 配置protobuf,编译
接着运行test.py遇到错误 加载不了tf,需要把库换成9.0
然后这里我才知道了annaconda虚拟环境的强大,比如conda下安装框架时会顺带安装对应版本的cuda,但是不知道安装在哪里,cudnn也没法放,还有虚拟环境下框架会自动选择对应的cuda,前面也说了cuda只是一个包
然后运行test.py成功

接着一开始按照网上说的进jupyter notebook运行demo,结果进去了发现内核错误,到现在也没解决,百度了才发现这个例程连缩进都不对具体我参考这个博文改的
https://blog.csdn.net/qq_34767784/article/details/90263482
然后我直接copy到pycharm里面运行了,要路径啥的,这个也不难,最后一步程序运行完了,结果不显示,万里长征差最后一步,我百度看了很多方法,很很多博客不一样,这些对我没用的我就不说了,你一百度也有,我就说我的解决办法,
在这里插入图片描述
D:\models\research\object_detection\utils\visualization_utils.py
下import matplotlib; matplotlib.use(‘Agg’) # pylint: disable=multiple-statements 这句里面的’Agg’改成’TkAgg’
然后出图,搞定
在这里插入图片描述

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值