![](https://img-blog.csdnimg.cn/20201014180756923.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch
waiting&fighting
我喜欢高效率和一劳永逸
展开
-
使用os和glob对样本进行分类存储
使用os和glob对样本进行分类存储 from glob import glob import os import numpy as np path = r'E:\Data\机器学习数据\new-dogs-vs-cats' rpath = r'E:\Data\机器学习数据\dogs-vs-cats\train' # 读取文件夹内的所有文件 files = glob(os.path.join(rpath, '*.jpg')) print(f'Total no of images {len(files)}'原创 2020-12-18 14:11:50 · 183 阅读 · 0 评论 -
使用pytorch实现resnet18
代码 from torchvision.datasets import ImageFolder from torchvision.transforms import transforms from torch.utils.data.dataloader import DataLoader import torchvision.models as models import torch.nn as nn from torch.cuda import is_available as is_cuda import原创 2020-12-18 16:13:12 · 2017 阅读 · 2 评论 -
样本生成
样本生成 def GenerateSamples(rPath, Illu, wPath): ''' :param rPath: :param Illu: :return: ''' with open(rPath, 'r') as fr: with open(wPath, 'w') as fw: line = fr.readline() count = 0 while lin原创 2020-11-11 16:32:47 · 257 阅读 · 0 评论 -
pytorch构建一个神经网络的步骤
定义数据类 数据集若不存在,则从预设网址下载数据到给定位置。并做数据预处理。 数据集若存在,做数据预处理。 定义网络层类 网络层可以有多种,因此可以有多个不同的网络层类。当定义框架类用到这个图层的时候,例如卷积层。就可以直接在框架类的构造函数里初始化一个网络层类的对象。 定义网络框架类 _init_函数,forward函数 定义train和test函数 train函数用于网络训练,test函数用于网络测试。 注意:要记录训练和测试的结果,并写入指定文件和出图。 ...原创 2020-05-15 17:35:24 · 314 阅读 · 0 评论