- 博客(26)
- 收藏
- 关注
RSS订阅原创 在Pytorch中实现tf.extract_image_patches的功能
tf.extract_image_patches是tensorflow用来从一张图像中提取多个patches的，其实现抽取patches的方式请移步一篇知乎的文章，那里介绍的很清楚。最近尝试在pytorch中来实现tf.extract_image_patches的功能，具体代码如下：一、tensorflow中tf.extract_image_patches从Tensor中提取patches1、提取单通道张量import tensorflow as tf# 创建两个张量大小为6*6，且为单通道 (
2020-08-19 14:12:22
2885
2
原创 Python 将raw图pack成4通道及其逆运算unpack!
一般神经网络处理的数据都是数组，raw图数据读入时是单通道的数组，如(H, W);而基于raw数据的深度学习图像处理基本上都是将raw数据pack成4通道的数组，即每一个通道分别有 R, Gr, Gb, B像素值组成的，形如(4, H / 2, W / 2)；然后再送进神经网络进行运算学习的。下面是将raw数据(H, W)，pack成4通道形式(4, H / 2, W / 2)的方法，以及其反向操作的方法。一、将raw数据才成4通道：def pack_raw(raw): """ :pa
2020-08-03 19:02:37
2540
原创 pytorch 简单的tensor插值放大和缩小运算
放大2倍插值运算代码：def interpolate(raw_tensor): ''' 等效于F.interpolate(source, scale_factor=scale, mode="nearest") :param raw_tensor: (B, C, H, W) :return: (B, C, 2 * H, 2 * W) ''' b, c, h, w = raw_tensor.shape out = torch.zeros((b, c, h
2020-08-03 18:20:25
6423
原创 使用转置卷积和卷积分别实现 pytorch 中 nn.PixelShuffle 层的正运算和逆运算。
pytorch 中 nn.PixelShuffle 层主要是将Tensor的通道数降低4倍的同时将其分辨率扩大2倍，但整个过程是不会改变Tensor中的数值的。简单理解就是，nn.PixelShuffle 层的输入和输出是相同的。nn.PixelShuffle 层的具体运算请参见pytorch官方介绍，我主要是使用转置卷积来实现与 nn.PixelShuffle 层相同的功能；并可以使用卷积层实现其相反的功能，也就是它的逆运算。详细代码如下：一、使用转置卷积层实现nn.PixelShuffle 层的功能
2020-08-03 18:15:42
2754
3
原创 Pytorch 张量滤波中值替换
使用一个小尺寸的值都为1张量对一个大尺寸的张量进行中值滤波，将小尺寸张量对应卷乘大尺寸张量，并将卷乘得到的结果排序后将中值替换：import torchimport numpy as npdef medfilt(volume): kernel_size = torch.ones(3, 3) w, h = volume.shape[0], volume.shape[1] for i in range(h- 3 + 1): for j in range(w - 3
2020-07-17 19:02:58
2549
2
原创 在pytorch中取一个tensor的均值，然后该张量中的所有值与其对比！
Pytorch中的Tensor的shape是(B, C, W, H)，对该tensor取均值并与所有值做对比代码如下：C, H, W = tensor.shape[1], tensor.shape[2], tensor.shape[3]for c in range(C): mean = torch.mean(x[0][c]) for h in range(H): for w in range(W): if x[0][c][h][w] >= mean: x[0][c][h][w] =
2020-07-17 17:11:05
4551
1
原创 YUV视频序列转RGB图像
# %% md## yuv视频转为rgb图像""""将一个文件夹中包含不同场景的文件夹下的YUV转为图像，对应保存到另外一个文件夹，相同的该文件夹下有着代表不同场景的文件夹，这些文件夹下的图像就是由YUV转来的"""# %%import osimport cv2import numpy as npdef yuv2bgr(video_dir, height, width, startfrm): """ :param filename: 待处理 YUV 视频的名字
2020-07-17 15:01:18
697
原创 Python 批量将raw数据或图片一一存成.npy文件
深度学习神经网络训练读入的都是numpy数组，一般直接读取图片在做预处理送入模型训练时很浪费时间的，我们可以提前将数据转成.npy格式的文件，并且数据的预处理也可以在这里完成。那么，将预处理好的数据直接送入网络就可以节省很多的时间了。关于下面有关raw数据读取看不懂的可以查看我的另外一篇专门敢于读raw数据转png图片的博客，里面有解释。# -*- coding:utf8 -*-import numpy as npimport cv2import osdef oneimg2npy_(roo
2020-07-17 14:54:35
2941
原创 Python raw格式图像转png格式
raw数据的读取有rawpy，numpy和逐元素读取等方法，一般在raw数据上进行深度学习神经网络的研究需要先将raw转换成数组方式才能输入到模型中进行训练。先介绍两种读取raw数据为数组并将之转为png图片的方法：1、使用rawpy来读取raw数据注意raw数据有很多种不同后缀名，如：.raw、.dng、.arw等。raw数据直接来源于sensor，不同公司的sensor是不一样的，他们的数据后缀名也就不一样了。下面以.dng为例，其他也一样的：import numpy as npimport i
2020-07-16 20:19:36
6043
7
原创 Linux终端杀死所有Python进程！
杀死所有Python进程的命名：ps aux|grep python|grep -v grep|cut -c 9-15|xargs kill -9
2020-07-09 19:32:02
2384
2
原创 Linux系统命令端复制文件夹或文件
复制文件夹或文件命令：sudo -r cp /var/www/html/AAA /var/www/BBB（AAA复制到BBB中）
2020-07-09 19:28:19
575
1
原创 Pytorch 结合Sobel滤波的卷积操作来提取图像的轮廓图！
在pytorch中实现将sobel算子和卷积层结合来提取图像中物体的边缘轮廓图，如下代码是卷积执行soble边缘检测算子的实现：import torchimport numpy as npfrom torch import nnfrom PIL import Imagefrom torch.autograd import Variableimport torch.nn.functional as Fimport cv2def edge_conv2d(im): # 用nn.Conv2
2020-07-09 14:57:50
7178
7
原创 Python 神经网络 使用numpy检查数据中是否有脏数据！
在神经网络训练之前，需要检查核对数据。对于庞大的数据集，如果其中存在少量的脏数据，人为检查是很难查到的。因为输入到神经网络的都是Numpy数组，所以可以使用numpy的工具来检查自己的数据中是否存在脏数据，代码如下：import nump as np # 下面的程序嵌套在数据读取的循环内if True in (img == np.nan): # 打印第idx个数据是脏数据 print("this is a bad data:{}".format(idx)) # 打印这个脏数据值的位置 pri
2020-06-23 18:39:21
704
转载 linux 将一个服务器上的文件或者文件夹复制到另一台服务器上
使用scp将一个Linux系统中的文件或文件夹复制到另一台Linux服务器上复制文件或文件夹（目录）命令：一、复制文件：（1）将本地文件拷贝到远程scp 文件名 用户名@计算机IP或者计算机名称:远程路径本地192.168.1.8客户端scp /root/install.* root@192.168.1.12:/usr/local/src（2）从远程将文件拷回本地scp 用户名@计算机IP或者计算机名称:文件名 本地路径本地192.168.1.8客户端取远程服务器12、11上的文件s
2020-06-20 08:42:02
2914
原创 双系统Windows + Ubuntu下，在Ubuntu系统中打开硬盘的挂载命令！！！
电脑安装双系统后，在Ubuntu系统时会出现打不开Windows系统中的D盘等情况时，需要先进行挂载才可以使用，挂载命令是：·sudo ntfsfix /dev/sda*注意：是你在点击打开D盘或其他盘时，打不开后会给你的信息，每个盘对应的数字是不一样的，只要跟着提示修改*就行。...
2018-07-21 16:37:48
2164
原创 Ubuntu系统配置TensorFlow/Theano+ Keras/PyTorchGTX1080Ti+Cuda-8.0+Cudnn-8.0-v6.0+OpenCv3.1+Python2/3
一、显卡驱动的安装Ubuntu可以自动安装驱动，在System Settings..中进行安装，如图： 二、安装cuda 8.01. cuda 8.0下载 现在的cuda9.0及以上版本都已进出来了，官网（https://developer.nvidia.com/cuda-downloads）的下载已经默认下载9.0版本找不到8.0版的了;我自己把cuda 8.0存...
2018-06-01 09:52:15
501
转载 使用OpenCV处理深度估计相关的数据集的通道数及位深的实现注意点！
import cv2cv2.imread("filename", flags)对于cv2，imread的关于通道数和位深的flags的四种选择：# 不进行转化，比如保存为了16位的图片，读取出来任然是16位。IMREAD_UNCHANGED = -1# 进行转化为灰度图，比如保存为了16位，读取出来位8位。IMREAD_GRAYSCALE = 0# 进行转化为RGB三通道...
2018-05-22 19:41:35
1097
原创 Ubuntu命令端下载软件临时换源！
在命令端安装或更新一些软件时，可能出现下载速度很慢，这时采用临时换源的方法可以极提高很多有很多的源可供选择，我经常使用的是豆瓣源或者清华源。个人觉得这两个源挺稳定和快的，下面使用豆瓣源为例给出命令：·sudo pip3 install xx -i https://pypi.douban.com/simple注：xx代表你要安装软件的名称及版本，如：tensorflow-gpu==1...
2018-05-22 19:26:47
8884
原创 Ubuntu系统在终端查看一些软硬件版本或文件大小的指令！
1、查看显卡的相关信息命令：·nvidia-smi2、查看opencv版本的命令：·pkg-config --modversion opencv3、查看python的版本的命令：·python --version4、查看cuda的版本的命令：·cat /usr/local/cuda/version.txt5、查看cudnn的版本的命令：·cat ...
2018-05-22 19:06:54
1019
转载 ImportError: cannot import name 'main'错误解決办法！
一、使用pip出现： Traceback (most recent call last): File “/usr/bin/pip3”, line 9, in from pip import mainImportError: cannot import name ‘main’二、解决方法：输入以下命令重新安装pip。sudo python3 -m pip uni...
2018-05-22 18:45:34
5786
转载 Unbuntu 下批量图片格式转换的方法！！！
以.png格式的图片与.jpg格式的图片之间的相互转换为例：1、从PNG转换到JPG，在终端输入如下命令：·ls -1 *.png | xargs -n 1 bash -c 'convert "$0" "${0%.png}.jpg"'·rm -rf *.png注：第二条指令在转换图片的路径下删除.png图片，第一条指令在转换成功后不会自动删除原有的.png格式的图片。2、 从JP...
2018-05-22 15:55:57
1076
原创 Ubuntu 自带输入法fcitx配置！！！
Ubuntu配置fcitx输入法 1、 在命令终端输入：sudo apt-get install fcitx-table-wbpy注：wb：五笔输入法；py：拼音输入法2、在键盘输入方式系统处把IBus替换为fcitx，如图： 3、重启电脑 4、从Ubuntu 右上角的小键盘图标中打开，配置选择五笔或拼音，如图： 点击+号，进入选择，如图： ...
2018-05-21 16:52:07
24159
原创 Ubuntu在无法进入图形界面，只能在TTY界面需要挂载以及查看U盘或移动硬盘内容的方法！
一、首先运行命令：·sudo fdisk -l查看磁盘信息，可以得到其路径，如“/dev/sdb 1”，磁盘格式为fat 32。二、运行挂载命令：·sudo mount -t vfat /dev/sdb1 /media注：若U盘为NTFS格式的话，挂载命令为：·sudo mount -t nfs /dev/sdb1 /media三、运行命令：·cd /media进入挂载U盘的主目录，运行：·ls查...
2018-05-19 18:15:57
2404
原创 Fast RCNN的训练与测试
1.准备工作1.1软件准备首先，需要安装Caffe和pycaffe。caffe原作者网页：//caffe.berkeleyvision.org/installation.html注意：必须在Makefile.config配置文件中打开Python层支持。# In your Makefile.config, make sure to have this lineuncomm...
2018-03-12 16:46:17
5039
8
原创 Ubuntu 16.04的caffe环境配置：cuda 8.0+cudnn 8.0+opencv3.1.0 + python2.7 + matlab2016b + blas(OpenBlas)
本文是搭建用caffe来进行深度学习的所需的环境配置方法的介绍;具体配置内容如标题。在安装之前希望读者先阅读一遍事先将需要下载安装的东西提前都下载好，对于matlab的版本可以根据个人需求自行修改。caffe的环境配置网上有很多的教程，该篇文章是本人根据网上的一些教程和自己的一些经验作出的一些总结：一、显卡驱动的安装;方法一：1、查看显卡型号：·lspci | grep VGA...
2017-10-15 14:27:40
1397
原创 Ubuntu16.04 下Matlab2014a安装教程
Ubuntu 16.04下Matlab的安装，情况多变。新手入坑，我系统的需求和网上其他教程的方法有些出差，改了好久才装成功。同时将一些问题和注意点提出来了，希望能够帮助遇到和我类似问题的人！
2017-08-25 20:43:39
4569
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人