
深度学习
部分深度学习记录
SimpleZihao
听君一席话,如听一席话
展开
-
深度学习的一些小记录
里面有一部分是摘录1. dim函数的部分理解tf.nn.functional.softmax(x,dim = -1)中的参数dim是指维度的意思dim = 0, 对每一维度相同位置的数值进行softmax运算dim = 1, 对某一维度的列进行softmax运算dim = 2, 对某一维度的行进行softmax运算dim = -1, 对某一维度的行进行softmax运算2. enumerate的使用:将可迭代对象如字符串、列表、元组、字典、集合,组合成索引+数据格式输出,第一个参数为可迭代原创 2021-05-31 10:53:09 · 276 阅读 · 0 评论 -
pytorch训练cifar10数据集预测图片
# -*- coding: utf-8 -*-# @Author : Fuzihao# @File : Image_classification.py# @Project: pythonProject14# @CreateTime : 2021/4/1 13:56:31import timeimport cv2import torchimport torchvisionfrom PIL import Imageimport torch.nn as nnimport torchvisi原创 2021-04-16 17:26:21 · 390 阅读 · 2 评论 -
深度学习入门之卷积神经网络CIFAR10数据集训练(pytorch)
深度学习入门之卷积神经网络训练CIFAR10数据集import torchimport torchvisionimport torch.nn as nnimport torch.nn.functional as Ffrom torch.utils.data import DataLoaderimport torchvision.transforms as transform# 加载数据集 并归一化transform = transform.Compose( [transform原创 2021-04-04 19:41:00 · 580 阅读 · 0 评论 -
深度学习入门之卷积神经网络训练MNIST数据集(pytorch)
深度学习练手之卷积神经网络训练MNIST数据集(pytorch)import timeimport torchimport torchvisionimport torch.nn as nnimport torch.nn.functional as Ffrom torch.utils.data import DataLoaderimport torchvision.transforms as transformsstart_time = time.time()# 数据预处理# !!!原创 2021-04-04 19:18:26 · 544 阅读 · 0 评论