合集-CNN案例:手写数字

摘要:本文基于python-pytorch框架,编写CNN网络,并采用CNN手写数字数据集训练、测试网络。 网络的构建 以LeNet-5 网络为例 类定义 首先先了解一下网络的最基本框架 一般而言,首先创建一个类class,创建时,继承nn.Module父类,注意,在该类的构造函数中__init__中,显 阅读全文
posted @ 2024-07-21 13:08 SXWisON 阅读(100) 评论(0) 推荐(0) 编辑
摘要:数据集下载 这一部分比较简单,就不过多赘述了,把代码粘贴到自己的项目文件里,运行一下就可以下载了。 from torchvision import datasets, transforms # 定义数据转换,将数据转换为张量并进行标准化 transform = transforms.Compose( 阅读全文
posted @ 2024-07-22 15:26 SXWisON 阅读(123) 评论(0) 推荐(0) 编辑
摘要:import torch import torchvision.transforms as transforms from PIL import Image import numpy as np import cv2 import matplotlib.pyplot as plt from net 阅读全文
posted @ 2024-07-22 15:34 SXWisON 阅读(37) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示