原创 通過PyTorch實現風格遷移

一.環境及數據集準備 pytorch 1.1.0 torchvision 0.3.0 cuda 9.0 數據集用的是COCO2014的train2014訓練集,使用ImageNet也可以 需要用到在ImageNet上預訓練好的VG

原创 通過PyTorch用DCGAN生成動漫頭像

數據集 數據集我們用AnimeFaces數據集,共5萬多張動漫頭像。 https://pan.baidu.com/s/1eSifHcA 提取碼:g5qa 要把所有的圖片保存於data/face/目錄下,後邊用ImageFolder

原创 PyTorch對DCGANs網絡的實現

原文鏈接:https://github.com/L1aoXingyu/code-of-learn-deep-learning-with-pytorch/blob/master/chapter6_GA

原创 PyTorch對ResNet網絡的實現解析

PyTorch對ResNet網絡的實現解析 我在博客園的這篇文章 1.首先導入需要使用的包 import torch.nn as nn import torch.utils.model_zoo as model_zoo # 默認