![](https://img-blog.csdnimg.cn/20201014180756918.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
记录
iviyandyssg
这个作者很懒,什么都没留下…
展开
-
【踩坑2】cv2.resize的意外之喜--降维
cv2.resize()实现了对图像的缩放具体的参数和使用可以参考cv2.resize()_一行一生-CSDN博客_cv2.resize但是cv2.resize()有个很多教程都没提到的小坑,如果输入的图像是channel是1(也就是维度维[size1,size2,1]的图像,和灰度图像又不一样,cv2把灰度图像读入维度是[size1,size2]),resize之后维度会直接变为[newsize1,newsize2]而不是[newsize1,newsize2,1]之前写代码的时候为了能同时用于原创 2021-10-12 08:56:11 · 410 阅读 · 1 评论 -
【踩坑1】指定GPU
在服务器上有多块gpu时,如果不指定就会默认用cuda0,如何指定用哪一块GPU。1、直接在代码的开头指定import osos.environ["CUDA_VISIBLE_DEVICES"] = "6"2、用.to()或者.cuda()指定device = torch.device("cuda:6" if torch.cuda.is_available() else "cpu") # 单GPU或者CPUmodel.to(device)net = net.cuda(6)# .原创 2021-09-22 21:13:42 · 1110 阅读 · 0 评论 -
【深度学习2】DataLoader的collate_fn参数
参考文章:https://blog.csdn.net/weixin_42028364/article/details/81675021collate_fn是torch.utils.data.DataLoaderd的参数之一,用来定义如何取样,可以不赋值,也可以自己定义函数来实现自己想要的功能。比如:def collate_fn(data): images,captions, sentence_num, max_word_num, image_id = zip(*data) im原创 2021-09-01 14:59:22 · 686 阅读 · 2 评论 -
【记录2】Python2代码转Python3代码 -- 2to3.py使用
Python2的代码在Python3的环境下可能会报错。 比如Python2中的输出print "Hello World!"在Python3中应该写为print("Hello World!"),否则代码会报错。 这个问题可以通过Python的内置工具2to3.py(也可能是2to3-script.py)解决,2to3可以自动把Python2的代码转化为Python3的代码。 首先要找到自己的2to3.py/2to3-sc...原创 2021-08-31 17:34:40 · 211 阅读 · 0 评论 -
【记录1】常用的预训练pytorch模型下载地址
ResNet系列在ImageNet上的预训练模型 'resnet18': 'https://download.pytorch.org/models/resnet18-5c106cde.pth', 'resnet34': 'https://download.pytorch.org/models/resnet34-333f7ec4.pth', 'resnet50': 'https://download.pytorch.org/models/resnet50-19c8e357.pth'原创 2021-07-06 16:29:44 · 172 阅读 · 0 评论