Pytorch入门手写数字

#导入需要的包 import numpy as np import torch from torch import nn from PIL import Image import torchvision import matplotlib.pyplot as plt import os from torchvision import datasets, transforms,utils Step1:准备数据。 transform = transforms.Compose([transforms.ToTensor(), transforms.Normalize(mean=[0.5],std=[0.5])]) train_data = datasets.MNIST(root = "./data/", transform=transform, train = True, download = True) test_data = datasets.MNIST(root="./data/", transform = transform, train = False) print(len(train_data)) print(len(test_data)) 60000 10000 train_data 的个数:60000个训练样本 test_data 的个数:10000个训练样本 train_loader = torch.utils.data.DataLoader(train_data,batch_size=128, shuffle=True,num_workers=2) test_loader = torch.utils.data.DataLoader(test_data,batch_size=128, shuffle=True,num_workers=2) print(len(train_loader)) print(len(test_loader)) 469 79 加载到dataloader中后,一个dataloader是一个batch的数据...

April 9, 2021 · wuyangzz

Jupyter中matplotlib中文乱码问题

下载字体 首先下载中文字体,链接如下。 字体下载链接 win中下载号以后可以直接安装字体 修改matplotlibrc文件 matplotlibrc文件的位置如下: D:\Anaconda3\Lib\site-packages\matplotlib\mpl-data (1)删掉 font.family前面的 # 并改为SimHei (2)删掉 font.sans-serif 前面的 # 需要将刚才下载的SimHei添加到其中,如红色方框所示。 保存退出! 把这个路径下的文件都删除(tex.cache和fontlist-.v310josn) c://用户//用户名//.matplotlib 重新启动jupyter

March 26, 2021 · wuyangzz

图片加噪声

图片加噪声 图像噪声分类 1.加性噪声 f(x,y)=g(x,y)+n(x,y) 一般是图像传输信道噪声和CCD摄像机图像数字化过程中产生 2.乘性噪声 f(x,y)=g(x,y)*n(x,y) 一般由 胶片中颗粒 飞点扫描图像噪声 电视扫描光栅等原因造成 3.量化噪声 模拟到数字产生的差异 量化中的误差 图像噪声模型 1.高斯噪声 (Gaussian noise) 最广泛。传感器非正常环境下产生,电子电路中噪声。 高斯分布 2.脉冲噪声 (Impulsive noise) 双极脉冲:椒盐脉冲,尖峰噪声 散粒噪声 盐噪声:随机的白色像素点 胡椒噪声:随机黑色像素点 3.瑞利噪声 (Rayleigh noise) 4.伽马(爱尔兰)噪声 (Gamma noise) 5.指数噪声(Exponential noise) 6.均匀噪声(Uniform noise) 去噪效果评价算法 1.SNR [信噪比] 计算图像自身的信噪比 输入为一幅图片 2.PSNR [峰值信噪比] 计算两个图像之间的相似度 去噪后的图片和原图做比较 3.SSIM [结构相似性] 衡量两幅图像相似度 Code import numpy as np import copy import cv2 import random import skimage.metrics import matplotlib.pyplot as plt def GaussianNoise(srcImg,percent,sigma,means=0,greyscale=256): """ 为灰度图像添加 高斯噪声 :param srcImg: 源图像 :param percent: 噪声百分比 :param sigma: 高斯的标准差 :param means=0: 高斯的均值 默认为0 :param greyscale=256: 灰度图像的度 默认为256 """ (h,w)= srcImg....

March 26, 2021 · wuyangzz