pytorch DataLoader中的参数:collate_fn函数

def default_collate(batch): r"""Puts each data field into a tensor with outer dimension batch size""" error_ms...

2019-06-30 19:50:14

阅读数 17

评论数 0

pytorch DataLoader源码解析

#文章源码来自:pytorch源码。 一、DataLoader的源码: class DataLoader(object): r""" Data loader. Combines a dataset and a sampler, and prov...

2019-06-30 19:45:19

阅读数 14

评论数 0

pytorchCNN网络权重

ssh://user2@219.216.65.100:6002/home/user2/anaconda3/envs/chenyafang/bin/python -u /home/user2/yd/lungcancer_0521/DSB2017-master/add_by_yd/test_net_s...

2019-05-29 20:33:07

阅读数 39

评论数 0

Ubuntu16.04中vim支持python

1. 查看现在系统中的vim支持哪个版本的Python vim --version 2."+"代表支持python3 3.要想支持python2: sudo apt-get install vim-nox-py2 4. 安装安成后再次查看一下是否支持: v...

2019-05-26 15:01:13

阅读数 15

评论数 0

windows下gist.github.com无法访问

1.管理员方式打开:记事本 2.记事本打开:C:\Windows\System32\drivers\etc\hosts 3.最后一行添加:192.30.253.118 gist.github.com 4.保存,大功告成。

2019-05-25 09:37:16

阅读数 6

评论数 0

pytorch 中tensor在CPU和GPU之间转换,以及numpy之间的转换

1. CPU tensor转GPU tensor: cpu_imgs.cuda() 2. GPU tensor 转CPU tensor: gpu_imgs.cpu() 3. numpy转为CPU tensor: torch.from_numpy( imgs ) 4.CPU ten...

2019-05-24 16:59:37

阅读数 362

评论数 0

pytorch expand()函数

转自https://pytorch-cn.readthedocs.io/zh/latest/package_references/Tensor/#expandsizes 1. 返回tensor的一个新视图,单个维度扩大为更大的尺寸。 tensor也可以扩大为更高维,新增加的维度将附在前面。 扩大...

2019-05-17 16:41:24

阅读数 115

评论数 0

判断pytorch是否支持GPU加速

print torch.cuda.is_available()

2019-04-30 09:58:18

阅读数 508

评论数 0

pytorch中Tensor的数据类型

1. 32位浮点型:torch.FloatTensor a=torch.Tensor( [[2,3],[4,8],[7,9]], ) print "a:",a print "a.size():",a.size() print "a.dtype:...

2019-04-30 09:55:53

阅读数 175

评论数 0

matplotlib画图,去掉坐标轴刻度,同时去掉坐标轴

1. 去坐标轴: plt.axis(‘off‘) 2.去刻度: plt.xticks([]) plt.yticks([]) 3.执行上述操作时,需要将其置于 plt.show() 之前,plt.imshow() 之后。

2019-04-24 16:23:39

阅读数 395

评论数 0

SimpleITK读取一个serie的文件

test_imgs_path=os.path.join( "tmp_path" ) reader=sitk.ImageSeriesReader() series_ids=reader.GetGDCMSeriesIDs(test_imgs_path) filenames=read...

2019-04-22 22:24:04

阅读数 33

评论数 0

SimpleITK保存mhd+raw和mha文件

1.使用SimpleITK将同一个序列的dicom保存为mhd+raw文件时,需要将dicom图像的InstanceNumber当做索引来存储,不能按照SliceLocation属性的大小来存储(会出错)。 2.吐过dicom图像的InstanceNumber,那么取InstanceNumber...

2019-04-21 11:06:46

阅读数 167

评论数 0

深度学习程序终止,nvidia显卡仍然被占用问题

1.查看占用显卡的进程,接着运行程序的时候,会提示显卡内存不足。 fuser -v /dev/nvidia* 2.使用kill pid,终止占用nvidia显卡的进程即可。 kill pid

2019-04-10 17:19:05

阅读数 207

评论数 0

图像处理用到concatenate操作时,size不匹配

图像处理用到concatenate操作时,size不匹配,像在unet的skip connection中, 1.第一种方式,使用padding 0填充。 2.如果不使用padding填充,很多情况是由于图像尺寸上采样、下采样之后尺寸不匹配,是有奇数、偶数造成的。避免这种情况的做法是:假设下采样...

2019-04-06 10:36:08

阅读数 81

评论数 0

skimage中measure方法

from skimage.measure import label,regionprops regionprops方法与label方法使用更好。 可以标记出每个联通区域。 label方法将图像的每个联通区域使用不同的像素标记出来,regionprops计算每个联通区域的属性(包含area(联...

2019-04-04 22:38:21

阅读数 192

评论数 0

matplotlib画直方图

1. plt.hist( self.pixels,np.linspace( 0,0.05,20 ) ) plt.xlabel( " range " ) plt.ylabel( " count " ) plt.show() 第一个参数代表所有需要分布的数...

2019-04-03 11:06:22

阅读数 44

评论数 0

pycharm运行和命令行运行代码的区别

1.pycharm运行py文件会自动将当前运行py文件的工程目录加载到sys.path中。 2.命令行运行py文件值加载显式import的包和文件,导致在命令行下运行py文件时会提示部分模块找不到。 3.在2中,如果找不到文件出错的话,使用: import sys sys.path.app...

2019-04-01 09:07:44

阅读数 181

评论数 0

numpy中的数据类型转换

1.两种方式:array.dtype=np.uint8 或是 array=array.astype( np.uint8 ) 2.使用array=array.astype( np.uint8 )的时候必须astype后赋值,只是用array.astype( np.uint8 ) 是不会改变数据类型...

2019-03-29 16:01:07

阅读数 130

评论数 0

Python生成固定位数的id,不足的用0填充

第一种方法: a=range( 10 ) print "{0:06d}".format( a[0] ) 第二种方法: a=range( 10 ) print "%04d"%(a[0]) 第三种方法: a=range( 10 ) pri...

2019-03-26 22:24:20

阅读数 150

评论数 0

Ubuntu1604查看网卡驱动

1. lspci 2.或者是: lshw -c network

2019-03-23 09:23:37

阅读数 587

评论数 0

提示
确定要删除当前文章?
取消 删除
关闭
关闭