![](https://img-blog.csdnimg.cn/20190927151043371.png?x-oss-process=image/resize,m_fixed,h_224,w_224)
修仙之路:代码实现篇
000000
宁静致远*
这个作者很懒,什么都没留下…
展开
-
FCN跳跃结构实现图像分割
1 BagData.py import os import cv2 import numpy as np from torch.utils.data import Dataset # onehot编码 def onehot(n, data): msk = np.eye(n)[data] # (C, H, W) => (224, 224, 2) msk = np.array(list(map(lambda x: abs(x - 1), msk))).astype('float3原创 2020-05-25 13:54:02 · 1221 阅读 · 0 评论 -
Pytorch:手写数字识别——MNIST数据集
本文以MNIST数据集为基础,描述如何构建并训练一个神经网络。分为以下几个步骤: 数据下载及预处理 构建CNN网络模型 训练CNN 测试CNN 使用CNN进行预测 1 数据下载及预处理 # 下载并预处理mnist数据集 import os import torchvision download_mnist = False # 判断mnist数据集是否已下载 if not os.path.exists('../MNIST') or os.listdir('../MNIST'): .原创 2020-05-14 20:36:46 · 1198 阅读 · 0 评论