![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
TensorFlow
吃了亏还皮
向左向右总要再向前
展开
-
显卡、CUDA、TensorFlow、PyTorch等之间各个版本问题大总结
显卡、CUDA、TensorFlow、PyTorch这四个组件想要能够稳定的运行在一个电脑上有点不容易,由于其多样性的版本经常出现各种问题,下面讨论下如何让四个组件协调工作,只考虑Windows系统下。显卡显卡是个人计算机最基本组成部分之一,用途是将计算机系统所需要的显示信息进行转换驱动显示器。显示芯片是显卡的主要处理单元,因此又称为图形处理器(Graphic Processing Uni...原创 2020-05-02 01:40:30 · 6937 阅读 · 4 评论 -
训练YOLO v4模型时,xml格式转txt格式
YOLO训练时需要的标签数据是txt格式的而以前做Faster-RCNN的时候,标签文件是xml格式的特别当下载别人标记好的数据集时,很有可能提供的是xml格式的文件,那么如何将xml格式转为txt格式?代码如下:import xml.etree.ElementTree as ETimport pickleimport osfrom os import listdi...原创 2020-05-01 00:54:39 · 4755 阅读 · 6 评论 -
Window环境下运行YOLO v4目标检测算法
GitHub原代码https://github.com/AlexeyAB/darknet,测试环境WIn7+CUDA9.0+Python3.6+VS2015社区版首先去GitHub上下载原代码。原版需要的CUDA版本为10,我的电脑装的CUDA是9.0版本的,所以需要手动修改,进入到build\darknet下,打开darknet.vcxproj,修改两处(将10.0改为9.0),可以参考下图...原创 2020-04-30 17:15:57 · 4454 阅读 · 6 评论 -
TensorFlow-Slim库 No module named 'nets'问题
Window系统下运行TensorFlow-Slim库下代码报错:解决方案进入slim文件夹下原创 2020-04-09 18:29:13 · 1177 阅读 · 1 评论 -
Windows下安装Tensorflow-Slim(待续)
官方Slim网址:https://github.com/tensorflow/models/tree/master/research/slim/nets(1)验证slim库在使用slim之前,要测试本地的tf.contrib.slim模块是否有效,在命令行中输入如下命令:python -c "import tensorflow.contrib.slim as slim; eval =...原创 2020-04-07 21:47:16 · 813 阅读 · 1 评论 -
InternalError (see above for traceback): Blas SGEMM launch failed : m=5329.......
Tensorflow报错,找了很多资料,一直不好。后来终于解决了,原来是兼容性问题,我用的显卡是RTX2070,CUDA版本9。原始的CUDA程序有BUG,需要打补丁才能正常运行,补丁可以去官网找CUDA补丁网址...原创 2020-04-07 10:13:25 · 2171 阅读 · 3 评论 -
Anaconda管理多个版本的Python环境
Anaconda本身安装了Python3.5,为了运行一些源码,需要Python3.6,可以新建一个环境,实现两种版本的共存在cmd里输入conda info --envs,查看已经安装的环境,只有一个名为base的环境 输入conda create --name py36 python=3.6,新建一个名为py36的环境,Python版本号是3.6再次输入conda...原创 2018-11-11 17:27:00 · 343 阅读 · 0 评论 -
GAN简单TensorFlow源码
原文链接Generative Adversarial Nets in TensorFlow论文链接Generative Adversarial Nets不废话,源码如下,用于生成minist手写体数字:# -*- coding: utf-8 -*-# @Author: adrianna# @Date: 2017-07-12 10:47:57# @Last Modified ...原创 2018-10-27 20:11:18 · 620 阅读 · 0 评论 -
高性能平台使用指南
学校新建的高性能平台,我负责调试,整理这份指南,供师弟师妹们参考激活工作环境使用XShell登录,校内IP 地址192.168.211.248,端口22,输入账号密码,登录成功界面如下所示。然后激活TensorFlow工作环境,Python版本3.6,TensorFlow版本1.10cd /share/software/anaconda3/binsource activa...原创 2018-09-03 16:16:15 · 2573 阅读 · 1 评论 -
Capsules胶囊神经网络代码测试(2)——测试cifar10数据
原地址:https://github.com/Sarasra/models/tree/master/research/capsules运行必须要有GPU下载两个数据包,第一个是 cifar10 tfrecords,第二个是已经训练好的模型文件。Download and extract cifar10 binary version to $DATA_DIR/ from https://w...原创 2018-08-16 16:59:34 · 1338 阅读 · 1 评论 -
制作自己的TFRecord数据集
一直在做CNN图像分类,原来采用的图像读入方式是放到文件夹下,直接将数据加载进内存,然后再分batch输入网络进行训练,但是后来发现太占用内存了,加了新内存条还是不够用。查阅资料,原来使用TFRecords这种结构能够有效地节省内存空间。下面就来制作自己的数据集吧!找到10类电商图像,调整和mnist类似,单通道,28x28大小。数据路径为D:\data\TFRecordimport...原创 2018-08-14 18:14:14 · 607 阅读 · 0 评论 -
Capsules胶囊神经网络代码测试(1)——测试mnist数据
原地址:https://github.com/Sarasra/models/tree/master/research/capsules运行必须要有GPU下载两个数据包,第一个是 MNIST tfrecords,第二个是已经训练好的模型文件。Download and extract MNIST tfrecords to $DATA_DIR/ from:https://storage.go...原创 2018-08-14 11:42:54 · 3758 阅读 · 1 评论 -
【转】Tensorflow张量操作
原文地址『TensorFlow』张量拼接_调整维度_切片1、tf.concattf.concat的作用主要是将向量按指定维连起来,其余维度不变;而1.0版本以后,函数的用法变成:t1 = [[1, 2, 3], [4, 5, 6]]t2 = [[7, 8, 9], [10, 11, 12]]#按照第0维连接tf.concat( [t1, t2],0) ==> [[1, 2...转载 2018-08-01 19:33:54 · 398 阅读 · 0 评论