PyTorch
gentleman_zh
这个作者很懒,什么都没留下…
展开
-
PIL_深度估计的数据集处理
一、将原始数据处理成PyTorch能够加载的数据 imgs为原始的RGB图像,dpts为其对应的深度图。 _path.txt为加载图片的路径,每一行都是:/rgb0.png /depth0.png class FireWork_Dataset(torch.utils.data.Dataset): def __init__(self, data_...原创 2019-01-10 16:26:58 · 979 阅读 · 0 评论 -
pytorch_tensorboardX_打开记录文件
tensorboard --logdir logs// 会出现一个网址,把http://zhaohan:6006/中的zhaohan改成localhost方可正常使用https://www.jianshu.com/p/46eb3004beca无法打开网址的解决方案:https://blog.csdn.net/minione_2016/article/details/75089208...转载 2019-01-03 09:34:07 · 2883 阅读 · 1 评论 -
python_opencv和plt加载图像
OpenCv以彩色图片的形式加载的图片是BGR模式。但是在Matplotlib中,是以RGB的模式加载的。使用OpenCv保存深度学习数据集中的RGB图像需要交换通道才能正常使用。 for imgs, dpts in test_loader: if torch.cuda.is_available(): imgs = imgs.cuda() ...原创 2019-01-18 15:11:24 · 2389 阅读 · 0 评论