iviyandyssg-CSDN博客

原创【踩坑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

原创【深度学习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

原创【深度学习1】梯度剪裁

参考文章：https://blog.csdn.net/qq_40178291/article/details/100853237?utm_medium=distribute.pc_relevant_t0.none-task-blog-2%7Edefault%7EBlogCommendFromMachineLearnPai2%7Edefault-1.essearch_pc_relevant&depth_1-utm_source=distribute.pc_relevant_t0.none-ta...

2021-08-31 17:56:52 168

原创【记录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

原创【学习5】jupyter notebook 增加虚拟环境

jupyter notebook默认使用base环境，但我的torch装在虚拟环境中，于是需要把虚拟环境添加上。参考文章：https://www.jb51.net/article/184447.htm添加虚拟环境需要用的ipykernel这个包，如果无，第一步是装上ipykernelpip install ipykernel假设要添加的虚拟环境是XX，展示的名称是YYpython -m ipykernel install --name XX --display-name "YY"

2021-08-13 17:26:10 106

原创【学习4】LSTM语料词典的生成

参考https://zhuanlan.zhihu.com/p/399181261、首先，必须有一个待处理的文本数据集 json_dir = r'.\data' json_file ='IUdata_trainval.json' vocab_dir = r'.\data' vocab_file= 'IUdata_vocab.pkl' json_path = os.path.join(json_dir,json_file) vocab_path = os

2021-07-26 09:49:38 366

原创【学习3】一些NLP评价指标及其计算

参考1、https://zhuanlan.zhihu.com/p/2230487482、https://www.cnblogs.com/xinbaby829/p/6955687.html3、https://zhuanlan.zhihu.com/p/376395631、bleuBLEU (全称为Bilingual Evaluation Understudy)的意思是双语评估替补。尽管最开始作为翻译的指标而被发明，但bleu也可以用于image caption的评估。假如给定标准译文

2021-07-24 20:32:29 2375

原创【学习2】EndNode的使用

1、下载并安装这一步可以自己去官网下载（https://endnote.com/）正版或者自己找破解版和汉化版的资源。2、新建库首次打开应该是这样的，一片灰色，什么都没有此时【文件】->【新建】来新建一个文献数据库，英文版对应的操作是【File】->【New】新建成功以后就会弹出这个文献数据库的操作界面，之后再打开也是这个界面同时保存会生成这两个文件夹，之后导入的文献信息和PDF保存在此PS. 导入的文献多了内存是会变大的，不要挑内存不够的地.

2021-07-15 10:13:04 720

原创【记录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

原创【学习1】Pytorch部分载入参数的几种方法

参考1 https://blog.csdn.net/LXX516/article/details/80124768参考2 https://www.bilibili.com/video/BV1dA411g7Ao方法一仅仅修改了全连接层，比如将分类层fc的1000类输出改为10类输出的new_fcimport torchfrom torchvision.models import resnet18import torch.nn as nndef main(): out_featu

2021-07-06 16:19:29 994

iviyandyssg的博客