pytorch深度学习
ChaoMartin
这个作者很懒,什么都没留下…
展开
-
torch.load_state_dict()函数的用法总结
在Pytorch中构建好一个模型后,一般需要进行预训练权重中加载。torch.load_state_dict()函数就是用于将预训练的参数权重加载到新的模型之中,操作方式如下所示:sd_net = torchvision.models.resnte50(pretrained=False)sd_net.load_state_dict(torch.load('*.pth'), strict=True)在本博文中重点关注的是 属性 strict; 当strict=True,要求预训练权重层数的键值与新原创 2021-07-12 23:31:44 · 94536 阅读 · 23 评论 -
Pytorch--Numpy--Random中的随机种子函数: seed(number)用法总结
保持随机初始化的一致性 -- 随机种子函数seedtorch 和numpy中随机初始化的一致性torch.manual_seed(number)torch.cuda.manual_seed(number)torch.cuda.manual_seed_all(number)numpy.random.seed(number)random.seed(number)torch 和numpy中随机初始化的一致性torch.manual_seed(number)https://pytorch.org/docs/m原创 2020-10-29 09:04:09 · 1980 阅读 · 0 评论 -
非极大值抑制算法总结(NMS, soft-NMS)
NMS-非极大值抑制算法总结greedy-nms、soft-nms标准非极大值抑制 - NMS软阈值NMS - soft-NMSgreedy-nms、soft-nmsgreedy-NMS是最传统的(标准)的非极大值抑制算法,soft-NMS是前者的改进版本。下面将分别从两者的原理与代码实现方面进行分析总结。标准非极大值抑制 - NMS标准NMS算法的流程:1.首先对需要进行抑制的同类别bounding-boxes按照预测的置信度conf进行降序排序;2.取出排好序的ns_boxes中的第一项,原创 2020-10-28 23:20:43 · 2854 阅读 · 5 评论 -
Python中使用警告:UserWarning: loadtxt: Empty input file: “/home/***/**/*.txt“的解决方案
问题描述:UserWarning: loadtxt: Empty input file: "/home/zh/PytorchProjects/00-DataSet/COCO2014/labels/train2014/COCO_train2014_000000195266.txt这只是一个使用警告,并不影响使用。警告loadtxt()函数可能读入空文件。解决方案:import warningswith warnings.catch_warnings(): warnings....原创 2020-07-03 19:03:07 · 3482 阅读 · 0 评论 -
PyTorch自学系列之 pretrained model usage
首先为了尊重他人贡献,本博客参考链接为:https://blog.csdn.net/VictoriaW/article/details/72821329(感谢知识传递者)楼主使用预训练模型的初衷是:目标识别检测框架在不同的计算机视觉应用中性能的提升很大一部分原因是由于骨干网络的西能提升,为了使用更好的Backbone获取feature map,所以有必要学习如何使用更好的预训练模型。套路...原创 2019-11-14 19:59:54 · 871 阅读 · 0 评论