PyTorch
Yi-27
小白一个...
展开
-
通过PyTorch实现风格迁移
一.环境及数据集准备pytorch 1.1.0torchvision 0.3.0cuda 9.0数据集用的是COCO2014的train2014训练集,使用ImageNet也可以需要用到在ImageNet上预训练好的VGG16,模型文件为vgg16-397923af.pth二.原理 首先需要明白一点,深度学习之所以被称为“深度”,就在于它采用了深层的网络结构,网络的不同层学到的是图...原创 2019-11-06 21:41:32 · 4345 阅读 · 11 评论 -
通过PyTorch用DCGAN生成动漫头像
数据集数据集我们用AnimeFaces数据集,共5万多张动漫头像。https://pan.baidu.com/s/1eSifHcA 提取码:g5qa要把所有的图片保存于data/face/目录下,后边用ImageFolder就能直接读取到。模型模型我们选择DCGAN。#coding:utf-8from torch import nnclass NetG(nn.Module):...原创 2019-10-20 21:08:17 · 2108 阅读 · 3 评论 -
PyTorch对ResNet网络的实现解析
PyTorch对ResNet网络的实现解析我在博客园的这篇文章1.首先导入需要使用的包import torch.nn as nnimport torch.utils.model_zoo as model_zoo# 默认的resnet网络,已预训练model_urls = { 'resnet18': 'https://download.pytorch.org/models/re...原创 2019-07-30 20:54:47 · 541 阅读 · 0 评论 -
PyTorch对DCGANs网络的实现
生成对抗网络(GANs)是现在深度学习的热点之一,下面我们通过PyTorch实现深度卷积生成对抗网络(DCGANs),数据集使用最为经典的MNIST手写数据集。1.导入需要使用到的包import osimport torchimport torch.nn as nnfrom torch.autograd import Variablefrom torch.utils.data im...转载 2019-08-11 16:44:50 · 462 阅读 · 0 评论