Linux学习
darkeyers
这个作者很懒,什么都没留下…
展开
-
TensorFlow ValueError: Didn‘t find op for builtin opcode ‘RESIZE_BILINEAR‘ version ‘3‘
遇到上述问题,升级TF至2.3或以上版本原创 2021-06-04 10:18:27 · 1636 阅读 · 0 评论 -
Pytorch tochvision版本变化导致的ResNet FClayer的变化
两台深度学习环境的机器一台配备的是torchvision版本为0.3.0 另一台版本为0.2.1同样的resnet代码出现了报错,显示FClayer的维度不匹配。查找原因如下:在0.2.1版本的resnet官方版本中对于avgpool Layer的定义如下:self.avgpool = nn.AvgPool2d(7, stride=1)k=7,s=1.而0.3.0版本的定...原创 2019-12-17 15:29:11 · 482 阅读 · 0 评论 -
Centos7 重装英伟达显卡驱动+Cuda9.0+Cudnn7
机房断电导致已经配置好的Centos7深度学习机器突然运行不了Tensorflow。并且出现以下情况,运行nvidia-smi 显示貌似没有问题。但是其实在运行深度学习框架的时候运行不了。估计是显卡驱动坏了。重装吧。首先卸载显卡驱动。目前我知道的显卡驱动安装方式有三种1、CUDA(.run)下载以后安装带上显卡驱动;2、显卡驱动.run文件安装3、集成软件包安装(yum等)接下来介绍后两种安装方式...原创 2018-05-08 10:52:41 · 22414 阅读 · 0 评论 -
Tensorflow学习笔记——ObjectDetectionAPI学习使用
写在前面:在开始ObjDetectionAPI学习之前请确认几个事情。1、识别机器是否可以正常跑Tensorflow1.4版本以上的Demo (即jupyter notebook中那个RunAll的Demo)吐槽一下:Demo中在每一次识别前都要下载对应的模型,而已经下载以后再重新下载就会报错(Python2.7)。2、识别机器是Tensorflow-gpu版。当然如果偏要用CPU去跑也可以 只不...原创 2018-07-12 20:52:01 · 731 阅读 · 0 评论 -
Pytorch 网络对输入图像要求固定大小的处理方法
对一些在ImageNet数据集上预训练的model,在分类的时候往往是使用的Fully Connected Layer,且往往是Linear(2048,1000)。而我们的测试图像往往不同于训练时候的224*224*3 ,那么经过以下方法的处理,可以经过网络处理。import torchfrom torchvision import datasets, transformsimport ...原创 2019-07-24 22:16:44 · 9801 阅读 · 0 评论