
tensorflow
White_XZX
这个作者很懒,什么都没留下…
展开
-
【Tensorflow Error】TypeError: `pred` must be a Tensor, a Variable, or a Python bool.
使用ssd_mobilenet_v1_pets.config配置文件对ssd模型进行训练时,模型跑起来后出现如下错误:解决方法:在object_detection目录下找到ssd_mobilenet_v1_feature_extractor.py文件,修改109行: 将is_training=None修改为is_training=True即可。参考资料:https://github.com/te...原创 2018-06-13 10:55:46 · 1183 阅读 · 0 评论 -
【GPU】GPU状态显示和部分指标含义
首先打开C盘,找到NVSMI文件夹,找到nvidia-smi.exe,打开cmd,将其exe拖到命令提示符窗口,回车后显示GPU信息,相应的指标信息如下:GPU:GPU 编号; Name:GPU 型号; Persistence-M:持续模式的状态。持续模式虽然耗能大,但是在新的GPU应用启动时,花费的时间更少,这里显示的是off的状态; Fan:风扇转速,从0到100%之间变动; Temp:温度,...原创 2018-06-14 21:30:12 · 17174 阅读 · 4 评论 -
【tensorflow】常用函数(持续更新)
1. tf.nn.droupout()作用:防止过拟合介绍:点击打开链接一般含有两个参数,第一个参数为输入,第二个参数为keep_prob,控制过拟合,其是一个概率值,在测试时一般设置为1,即保留全部结果,不起作用;在训练和测试时,一般为0.5,让机器遗忘50%的结果,当学习速度过快时让机器减慢,遗忘得多一点;当测试集的准确率不理想时可以调低该值,重新训练增加达到理想效果的概率。...原创 2018-07-02 17:37:03 · 350 阅读 · 0 评论 -
【Python】批量修改图片文件名和xml文件信息
在使用tensorflow进行数据训练时,由于原图片文件名较繁琐,且由于根据原图片名生成的xml标签文件中生成了包含filename的标签属性,不利于后期测试训练效果,故通过Python代码对图片名和xml文件信息进行批量修改为由0开始的顺序递增文件名,以配合后期的测试工作,在此记录下。首先,这是图片名称和xml文件信息:图片文件名:标签文件名:xml信息:Python...原创 2018-07-23 22:44:40 · 6933 阅读 · 9 评论