摘要: # -*- coding: utf-8 -*- import torch from torch.utils.data import Dataset from torchvision import datasets from torchvision.transforms import ToTensor 阅读全文
posted @ 2025-07-24 16:39 daviyoung 阅读(39) 评论(0) 推荐(0)
摘要: # 定义神经网络模型 class NeuralNetwork(nn.Module): def __init__(self): super().__init__() self.flatten = nn.Flatten() # 将 1x28x28 展平为 784 self.linear_relu_sta 阅读全文
posted @ 2025-07-24 16:38 daviyoung 阅读(15) 评论(0) 推荐(0)
摘要: custom_dataset.py # -*- coding: utf-8 -*- from torch.utils.data import Dataset import os import pandas as pd from torchvision.io import decode_image c 阅读全文
posted @ 2025-07-24 11:58 daviyoung 阅读(21) 评论(0) 推荐(0)
摘要: # -*- coding: utf-8 -*- import torchvision import os import gzip import struct import numpy as np from PIL import Image import pandas as pd torchvisio 阅读全文
posted @ 2025-07-24 11:54 daviyoung 阅读(18) 评论(0) 推荐(0)
摘要: import torch from torch.utils.data import Dataset from torchvision import datasets from torchvision.transforms import ToTensor import matplotlib.pyplo 阅读全文
posted @ 2025-07-24 09:30 daviyoung 阅读(13) 评论(0) 推荐(0)
摘要: ✅ 一、什么是灰度图? 灰度图(grayscale image):指的是只有黑白灰颜色的图像。 每个像素只用一个值表示亮度(0 表示黑,255 表示白,中间是不同程度的灰)。 举例: 彩色图像:每个像素用 RGB 三个通道(红、绿、蓝) 表示。 灰度图像:每个像素只需要一个值。 ✅ 二、什么是单通道 阅读全文
posted @ 2025-07-24 09:29 daviyoung 阅读(92) 评论(0) 推荐(0)
摘要: import torch import numpy as np print(" tensor create: ") data = [[1, 2],[3, 4]] x_data = torch.tensor(data) print(f"x_data Tensor: \n {x_data} \n") n 阅读全文
posted @ 2025-07-24 09:28 daviyoung 阅读(6) 评论(0) 推荐(0)