![](https://img-blog.csdnimg.cn/20201014180756919.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
文章平均质量分 66
半岛铁子_
那就叫小胡吧
展开
-
二维卷积输出特征图的计算公式
- H:高度;- W:宽度;- N:batch-size的大小;- Cin:输入的通道数;- Cout:卷积后输出的通道数;- dilation:膨胀率,的默认值是1;- stride:步长,默认是1。原创 2022-10-28 17:33:52 · 2287 阅读 · 0 评论 -
【报错记录】AttributeError: ‘xxx‘ object has no attribute ‘module‘
今天在跑代码时,报出 `AttributeError: 'InpaintGenerator' object has no attribute 'module'` 的错误在保存模型时,如果电脑是单GPU,报的错误的话,只需要把中第一个参数的.moudle去掉即可。原创 2022-10-20 21:46:11 · 13347 阅读 · 10 评论 -
pytorch dataset实战----读取数据集
这段代码可以作为读取数据集的一个DataSet类的基础类,可以扩充进行修改,以后有类似需要可以拿过来修改。代码中涉及到torch.transforms、合并路径等知识点,我在代码中都进行了详细的注释,路径要对照着项目结构,如果自己用的话要根据项目结构去将相对路径改过来。原创 2022-10-20 17:28:35 · 1200 阅读 · 3 评论 -
【报错记录】----RuntimeError:An attempt........ if __name__ == ‘__main__‘: freeze_support()
RuntimeError: An attempt has been made to start a new process before the current process has finished its bootstrapping phase. This probably means that you are not using fork to start your child processes and you have forgo原创 2022-10-15 20:24:52 · 457 阅读 · 2 评论 -
【数据增强】--- 使用 Pytorch 批量裁剪文件夹里图片并保存
在上一篇博客使用Pytorch裁剪图片并保存中说明了如何使用Pytorch对单张图片进行裁剪并保存。在实际使用过程中可能会面临对于一整个文件夹里的数据集批量进行裁剪并以指定的文件名保存到某一路径之下,本文基于这种需求对其进行实现。上面给的示例中,文件夹中只有3张图片,但是只要上面代码码跑通,无论文件夹中有多少张图片都是可以处理的。同时,可以更根据上面的示例去用Pytorch提供的其他方法切割图片,实现数据增强。原创 2022-10-08 15:55:06 · 3082 阅读 · 7 评论 -
【数据增强】---- 使用 Pytorch 裁剪图片并保存
最近需要制作一个数据集,由于数据集的数量不够,而这些数据集中的单张照片很大,因此可以通过裁剪的方式进行数据增强,从而获取到更多的数据,扩充数据集。在这个过程中还要将裁剪得到的图片保存下来。原创 2022-10-07 16:08:00 · 5787 阅读 · 2 评论 -
对于CNN中卷积的一些理解
图像是由一个个的像素点组合起来的。一般的彩色图像都分为RGB三个通道。图像就像是一个高纬度的张量,其本质还是由一个个0-255的像素所构成的,对于图像的操作,相当于是对这些像素点的数值进行计算。原创 2022-09-27 22:15:14 · 1590 阅读 · 2 评论 -
Pytorch中使用显卡(GPU)训练模型
在进行深度学习训练模型时,对于计算量小一些的模型,是可以在CPU上进行的。但是当计算量比较大时,我们希望利用GPU并行计算的能力去加快训练的速度。原创 2022-09-27 22:00:16 · 6272 阅读 · 0 评论 -
Pytorch加载数据集(DataSet和DataLoader)
Pytorch对于数据集的操作使用 DataSet 和 DataLoader。__len__在加载数据时,一般使用Mini-Batch,原因有以下两个:(1)通过使用batch,可以在梯度下降中更新参数时通过随机梯度下降的方法,这可以帮助我们跨越部分鞍点。(鞍点会导致梯度为0)(2)通过将数据划分为batch,可以充分利用计算机并行计算的能力,加快计算的速度。原创 2022-09-27 21:39:26 · 1730 阅读 · 0 评论 -
用Pytorch实现一个线性回归
对于一个模型,都要将其构建成一个类,并且继承自Module,至少定义两个方法,一个是构造函数,另一个是forward()。而backward会由Module里自动根据计算图计算。然后就是不断的前馈—反馈—更新----前馈—反馈—更新最后使Loss收敛。原创 2022-09-26 20:53:10 · 383 阅读 · 0 评论 -
Pytorch学习笔记(三)如何用GPU训练模型
对现有模型的使用,修改,保存和加载,如何对模型进行训练,如何调用GPU进行训练,以及在没有GPU的情况下使用免费的GPU进行远程训练,对模型进行验证等问题。原创 2022-07-26 11:11:16 · 1269 阅读 · 0 评论 -
Pytorch学习笔记(二)神经网络的使用
其中,input是输入图像,kernel是卷积核的大小,stride是步长,padding是填充的距离。原创 2022-07-26 10:57:19 · 483 阅读 · 1 评论 -
Pytorch学习笔记(一)安装与常用函数的使用
dataloader相当于如何从dataset中取出数据,dataset数据集,batch_size每一组的最大数量,shuffle是否打乱,num_workers线程数量,drop_last是否舍弃最后不被整除的图片。如果进行绘制了y=2x,又绘制了3x,会自动拟合图像,解决方法是删除logs下的所有文件,重新运行代码,重新运行tensorboard--logdir=logs--port=6007。(2)help()每个工具是如何使用的,工具的使用方法。版本有问题,将其卸载重装。...原创 2022-07-26 10:46:59 · 738 阅读 · 0 评论 -
Anaconda虚拟环境下安装opencv报错的问题
网上也有说升级pip再重新安装的,还有说进入到\anaconda\envs\pytorch\Lib\site-packages进行安装的,但是没有解决我的问题。之后继续安装opencv,又会报Failedbuildingwheelforopencv-python的错,尝试了多种方法也没有得以解决。最后的最后,我直接指定了opencv的版本(比最新的版本低)进行安装,安装成功了。在anaconda中创建了pytorch的虚拟环境,需要安装opencv,使用命令。于是又开始在找其他解决方案。...原创 2022-07-22 13:55:28 · 3492 阅读 · 2 评论