Pytorch框架
pytorch框架的基本使用
Lyang-Never
Never -
展开
-
pytorch——关于GPU的一些知识
一、前言最近在学习pytorch框架,记录一些涉及到的知识点,方便后续查找和学习。二、主要内容内容可能有些散,初次学习,可能把握不好知识之间的连贯性和整体性,后续适当调整。关于CUDA的一些函数接口:torch.cuda.is_available() #查看系统GPU是否可以使用,经常用来判断是否装好gpu版的pytorchtorch.cuda.current_device() #返回当前设备序号torch.cuda.get_device_name(0) #返回第0号设备的name原创 2020-06-03 18:33:56 · 894 阅读 · 0 评论 -
Pytorch中的permute函数和transpose,contiguous,view函数的关联
一、前言在进行深度学习的过程中,经常遇到permute函数,transpose函数,view函数,contiguous函数等,他们起什么作用,之间又有什么联系呢?二、主要内容2.1、permute函数和transpose函数Tensor.permute(a,b,c,d, …):可以对任意高维矩阵进行转置。例子见下:In[1]: torch.randn(2,3,4,5).permute(3...原创 2020-03-16 17:25:33 · 2319 阅读 · 0 评论 -
Retinaface代码记录(五)(损失函数)
一、写在开头这次主要记录关于Retinaface的损失函数部分。下面是代码地址:Retinaface代码地址主要包括的脚本为:multibox_loss.pybox_utils.py也欢迎阅读其上一篇博客Retinaface代码记录(一)。可以帮助读者对本片博客可以有一个整体上的把握和理解。...原创 2020-03-08 18:37:42 · 2479 阅读 · 0 评论 -
Retinaface代码记录(四)(网络结构)
一、写在开头这次主要记录关于Retinaface的网络结构部分。下面是代码地址:Retinaface代码地址主要包括的脚本为:retinaface.py.pynet.py也欢迎阅读其上一篇博客Retinaface代码记录(一)。可以帮助读者对本片博客可以有一个整体上的把握和理解。...原创 2020-03-08 18:36:02 · 3136 阅读 · 9 评论 -
Retinaface代码记录(三)(默认框生成)
一、写在开头这次主要记录关于Retinaface的锚框生成部分。下面是代码地址:Retinaface代码地址主要包括的脚本为:prior_box.py也欢迎阅读其上一篇博客Retinaface代码记录(一)。可以帮助读者对本片博客可以有一个整体上的把握和理解。...原创 2020-03-08 18:33:47 · 1142 阅读 · 8 评论 -
Retinaface代码记录(二)(数据处理)
一、写在开头这次主要记录关于Retinaface的数据处理部分。下面是代码地址:Retinaface代码地址主要包括的脚本为:wider_face.pydata_augment.py也欢迎阅读其提纲介绍。二、主要内容下面主要import osimport os.pathimport sysimport torchimport torch.utils.data as da...原创 2020-03-08 18:28:32 · 2697 阅读 · 13 评论 -
Retinaface代码记录(一)(基于pytorch)
一、记录内容这次的需求是:在Faceboxes的基础上,参照Retinaface加入了关键点信息。同时学习下Retinaface的代码和pytorch框架。Retinaface代码地址Faceboxes代码地址...原创 2020-03-07 00:36:37 · 2617 阅读 · 4 评论 -
PyTorch中 transforms的二十二个方法
传送门转载 2020-03-03 23:38:24 · 272 阅读 · 0 评论