草庐IT

CV-Pytorch

全部标签

Pytorch模型如何查看每层输入维度输出维度

在PyTorch中,可以使用torchsummary库来实现对PyTorch模型的结构及参数统计的输出,其可以方便我们查看每层输入、输出的维度以及参数数量等信息。安装torchsummary库:pipinstalltorchsummary使用方法如下:importtorchfromtorchsummaryimportsummary#定义一个模型model=torch.nn.Sequential()#打印模型所有层的参数统计summary(model,(3,32,32))其中,model是需要查看的模型,(3,32,32)表示模型的输入维度,即C=3,H=32,W=32。运行后可以看到所有层输入

解决cv2没有face模块

人脸识别报错解决:AttributeError:module‘cv2.cv2’hasnoattribute‘face’人脸识别报错如下:AttributeError:module‘cv2.cv2’hasnoattribute‘face’问题描述代码中cv2没有face模块所以报错,看了很多博主都是卸载和重新安装opencv-python库和opencv-contrib-python都是不行的,下面参考一位博主的亲测有用recognizer=cv2.face.LBPHFaceRecognizer_create()解决方案:直接在自己的虚拟环境下安装opencv-contrib-python库,后

Pytorch Mac GPU 训练与测评

今天中午看到Pytorch的官方博客发了AppleM1芯片GPU加速的文章,这是我期待了很久的功能,因此很兴奋,立马进行测试,结论是在MNIST上,速度与P100差不多,相比CPU提速1.7倍。当然这只是一个最简单的例子,不能反映大部分情况。这里详细记录操作的一步步流程,如果你也感兴趣,不妨自己上手一试。加速原理苹果有自己的一套GPU实现APIMetal,而Pytorch此次的加速就是基于Metal,具体来说,使用苹果的MetalPerformanceShaders(MPS)作为PyTorch的后端,可以实现加速GPU训练。MPS后端扩展了PyTorch框架,提供了在Mac上设置和运行操作的脚

Pytorch Mac GPU 训练与测评

今天中午看到Pytorch的官方博客发了AppleM1芯片GPU加速的文章,这是我期待了很久的功能,因此很兴奋,立马进行测试,结论是在MNIST上,速度与P100差不多,相比CPU提速1.7倍。当然这只是一个最简单的例子,不能反映大部分情况。这里详细记录操作的一步步流程,如果你也感兴趣,不妨自己上手一试。加速原理苹果有自己的一套GPU实现APIMetal,而Pytorch此次的加速就是基于Metal,具体来说,使用苹果的MetalPerformanceShaders(MPS)作为PyTorch的后端,可以实现加速GPU训练。MPS后端扩展了PyTorch框架,提供了在Mac上设置和运行操作的脚

Opencv出现错误 cv2.error:OpenCV(4.4.0)C:\Users\appveyor\AppData\Local\Temp\1\pip-req-build-6lylwdcz\open

今天在使用cv2.warpPerspective()函数时出现下面错误:百度了一下出现这样的错误是有两种可能,1、路径中含有中文字符导致cv2读取保存不了文件。2、路径错误或者路径不存在。可以在保存文件的路径下加入这两段代码。Save_path为你需要保存的路径。自己仔细检查了代码发现的确是自己读取图片名称时少了文件名图片原名为202211151_two-tone_fivestar-green.png修改了读取图片代码就正常了。

轴承故障诊断经典模型pytorch复现(一)——WDCNN

论文地址:《ANewDeepLearningModelforFaultDiagnosiswithGoodAnti-NoiseandDomainAdaptationAbilityonRawVibrationSignals》—张伟我们要复现的论文是轴承故障诊断里比较经典的一个模型WDCNN,最近在看的很多论文都把WDCNN作为比较模型,但是只找到过tensorflow版本的源码且只有原始的WDCNN没有改进的WDCNN-AdaBN版本,而我自己又是用的pytorch,因此就打算自己复现一下。话不多说直接上代码。WDCNN:#!/usr/bin/python#-*-coding:utf-8-*-im

opencv导入图片时,cv2后面无法添加imread的问题的解决方法

当使用cv2导入图片的时候,发现cv2.imread()被标明高亮,无法使用imread,resize等方法 针对这个问题,解决方法如下:首先在我们的Python安装路径下面找到cv2(安装anaconda的也是如此,找到相应路径),  在这里可以看到一个.pyd文件,将这个文件直接复制到site-packages路径下面即可 之后再去看我们的代码,可以看见cv2后面可以加imread等方法了  

cv2.error: OpenCV(4.6.0) D:\a\opencv-python\opencv-python\opencv\modules\imgproc\src\color.cpp:182:

报错:原因:路径问题解决办法:1.检查图片路径中是否有中文,有则修改2.将图片路径中的“\”改为“\\” 问题成功解决!!如果还是报错可留言。 

cv::FileStorage::Impl::open Can‘t open file: ‘haarcascade_frontalface_default.xml‘ in read mode

globalD:\a\opencv-python\opencv-python\opencv\modules\core\src\persistence.cpp(505)cv::FileStorage::Impl::openCan’topenfile:‘haarcascade_frontalface_default.xml’inreadmode在实验opencv对人脸的识别检测时,出现了无法打开对应的haarcascade_frontalface_default.xml文件的问题,推测时路径的问题,将路径更改为相应文件绝对路径,解决问题,具体原因未知,欢迎讨论

下载cv2的方法(OpenCV)

pipinstallopencv-python==4.5.3.56--user-ihttps://pypi.tuna.tsinghua.edu.cn/simple大家可以复制上面的命令在CMD或者AnacondaPrompt中直接下载 我本人用的python版本是3.7版大家下载好后到按一下步骤下载好cv2即可运用1. 2.3.点击右上角处的+  4.在该搜索栏中搜索你要安装的cv2