基于pytorch对图像进行卷积操作

基于pytorch对图像进行卷积操作

1. 首先将需要的库准备好，博主使用的是python3.8，pytorch1.7.1

import cv2
import torch
import torch.nn as nn
import matplotlib.pyplot as plt
import numpy as np

2. cv2读取图片，并将其转换为(640,640,3)，可以不转换

img = cv2.imread(r'D:\Hooded_Cotton_Canvas_Anorakimg_00000001.jpg')
img = cv2.resize(img, (640, 640))

原图

3. 转换为tensor张量

input_tensor = torch.from_numpy(img).float().permute(2, 0, 1)
input_tensor = input_tensor.unsqueeze(0)

4. 定义卷积层，此处卷积层的输入通道数为 3，因为图像是 RGB 图像。卷积核的大小为