夜视计算无法分析波形(PyTorch应用程序)

import torch
import torchvision.transforms as transforms
from PIL import Image

# 定义图像增强变换
transform = transforms.Compose([
    transforms.ToTensor(),
    transforms.Lambda(lambda x: x.clamp(0, 1)),  # 确保像素值在0-1之间
    transforms.Lambda(lambda x: x * 2 - 1)  # 归一化到[-1, 1]
])

# 加载低光图像
image = Image.open('low_light_image.jpg')
image_tensor = transform(image).unsqueeze(0)

# 定义简单的卷积神经网络
class SimpleCNN(torch.nn.Module):
    def __init__(self):
        super(SimpleCNN, self).__init__()
        self.conv1 = torch.nn.Conv2d(3, 32, kernel_size=3, padding=1)
        self.relu = torch.nn.ReLU()
        self.conv2 = torch.nn.Conv2d(32, 3, kernel_size=3, padding=1)

    def forward(self, x):
        x = self.conv1(x)
        x = self.relu(x)
        x = self.conv2(x)
        return x

model = SimpleCNN()

# 前向传播
output = model(image_tensor)
output_image = transforms.ToPILImage()(output.squeeze(0))

# 保存增强后的图像
output_image.save('enhanced_image.jpg')