- 博客(10)
- 收藏
- 关注
原创 ipynb文件import ipynb文件
stack flow上说明了两种方法第一,%run xxx.ipynb第二,需要预先pip install import_ipynbimport import_ipynbfrom xxx import xxx_function两种方法有区别,其一是将xxx.ipynb运行了一遍,第二种是通常的import。比如当xxx.ipynb中定义了if__name__=='__ma...
2020-03-22 21:20:43 4561 1
原创 【学习笔记】关于volatile和0.4.0版本后的torch.set_grad_enabled()
volatile是用来在验证或者测试的时候将输入设置成volatile,这样后继几点全部都被设置成volatile,也就是不需要自动求导。0.4.0以后采用torch.set_grad_enabled()来替代这种用法但是torch.set_grad_enabled()在使用的时候是设置一个上下文环境,也就是说只要设置了torch.set_grad_enabled(False)那么接下来所...
2019-03-28 17:08:30 8442 1
原创 【学习笔记】pytorch数据读取__getitem__的顺序问题
往往形式上是:def __getitem__(self,idx):idx的范围是从0到len-1(__len__的返回值)但是如果采用了dataloader进行迭代,num_workers大于一的话,因为是多线程,所以运行速度不一样,这个时候如果在__getitem__函,数里输出idx的话,就是乱序的。但是实际上当线程数设置为1还是顺序的。即使线程数大于1,如果返回idx,并...
2019-03-22 20:10:00 9446 5
转载 【学习笔记】图片读取
import skimage.io as ioimage = io.imread(image_dir).astype('float32') / 255mask = io.imread(mask_dir).astype('float32')from PIL import Imageimage = Image.open(image_dir)plt.imread和PIL.Image...
2019-03-14 10:51:31 491
转载 【学习笔记】一些基础
建立dockersudo nvidia-docker run -ti --ipc=host --name=xxx --network=host -v /home/data/xxx:/workspace ce893numpy的transpose :https://blog.csdn.net/u012762410/article/details/78912667比如import...
2019-03-02 21:53:09 351
原创 【学习笔记】pytorch多gpu
ssh连服务器docker中运行。可以先查看gpu运行状态,找空的gpu的id,比如是0,2,5,9这四个gpuwatch -n 0.1 nvidia-smi代码中加入device = torch.device('cuda')model = nn.DataParallel(model.to(device))在运行得时候指定gpu即可(注意DEVICES最后有一个S,被这个...
2019-03-02 16:04:30 254
原创 【学习笔记】关于维度,unsqueeze,squeeze
squeeze和unsqueeze可以参考:https://blog.csdn.net/xiexu911/article/details/80820028图片来源:https://blog.csdn.net/flysky_jay/article/details/81607289维度是(2,3)维度是(2,1,3)我一般这样理解维度,第一维是去掉第一个...
2019-03-02 15:57:03 243
原创 使用华硕xtion 摄像头遇到的问题(ros系统、turtlebot)
我的电脑是Ubuntu 14.04.5,ROS 为indigo版本。我们最初的愿望是实现基于turtlebot的路径规划;但是在使用华硕xtion的过程中出现了问题。 在运行openni后能够出现RGB图像和点云的图片。这种情况下会有warnin,但是这并不影响使用;运行成功的终端显示如下:mason@mason-VM62:~$ roslaunch openni_launch op
2017-08-30 11:09:06 2847 2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人