PyTorch获取卷积核的数值和大小

在PyTorch中，你可以通过访问卷积层（Convolutional Layer）的`weight`属性来获取其卷积核（也称为滤波器或权重矩阵）的数值。`weight`是一个张量，它包含了所有卷积核的值，每个通道对应一组卷积核。

例如，如果你有一个二维卷积层，其形状可能是`(out_channels, in_channels, kernel_height, kernel_width)`。其中：
- `out_channels`是输出通道数（即新特征图的数量）
- `in_channels`是输入通道数（通常是原图像的RGB通道数）
- `kernel_height`和`kernel_width`分别是卷积核的高度和宽度

获取卷积核的大小，你可以直接看`kernel_height`和`kernel_width`的值。如果你想查看单个卷积核的值，可以使用索引来提取，例如`weight[0]`代表第一个卷积核。

以下是获取和打印卷积核的基本步骤：

import torch.nn as nn
from your_model import YourModel  # 假设YourModel包含一个卷积层

# 初始化模型并加载预训练参数
model = YourModel()
model.load_state_dict(torch.load('your_model_weights.pth'))

# 获取卷积层
conv_layer = model.conv_layer  # 替换为实际模型中的卷积层名称

# 获取卷积核数值和大小
weights = conv_layer.weight.data
kernel_size = (conv_layer.kernel_size[0], conv_layer.kernel_size[1])

print(f"卷积核数值: {weights}")
print(f"卷积核大小: {kernel_size}")