当前位置:   article > 正文

这个卷积神经网络的python代码出的问题是什么?怎么解决?_python 神经网络错误

python 神经网络错误

#1、打开图像,转化为灰度图
import numpy as np
import torch
from torch import nn
from torch.autograd import Variable
import torch.nn.functional as F
from PIL import Image
import matplotlib.pyplot as plt
im=Image.open('D:/1234.jpeg').convert('L')
im=np.array(im,dtype='float32')#将其转换为一个矩阵
plt.imshow(im.astype('uint8'),cmap='gray')
#2、将图像数据转换为tensor张量,并适配卷积输入的要求
#3、创建Pytorch二维卷积神经网络对象,(卷积在pytorch中有两种方式,一种是torch.nn.Conv2d(),一种是torch.nn.functional.conv2d())
im=torch.from_numpy(im.reshape((1,1,im.shape[0],im.shape[1])))

conv1=nn.Conv2d(1,1,3,bias=False)#定义卷积
#定义轮廓检测卷积核
sobel_kernel=np.array([[-1,-1,-1],[-1,8,-1],[-1,-1,-1]],dtype='float32')
sobel_kernel=sobel_kernel.reshape((1,1,3,3))#适配卷积的输入输出
#给卷积的kernel赋值
conv1.weight.data=torch.from_numpy(sobel_kernel)

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/空白诗007/article/detail/776244
推荐阅读
相关标签
  

闽ICP备14008679号