用pytorch 直接卷积一张图片的灰度图并生成轮廓

最新推荐文章于 2025-09-18 00:58:08 发布

原创

最新推荐文章于 2025-09-18 00:58:08 发布 · 2.5k 阅读

8 ·

CC 4.0 BY-SA版权

本文档展示了如何使用PyTorch对灰度图像进行卷积操作，以实现边缘检测。通过加载图像，将其转换为张量，定义卷积层并应用自定义的轮廓检测算子，最后显示了卷积处理后的图像结果。

用pytorch 直接卷积一张图片.


import torch.nn.functional as F
import torch
import torch.nn as nn
from torch.nn.modules.conv import Conv2d 
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 
import cv2

torch.set_default_tensor_type(torch.DoubleTensor)

img = cv2.imread(