Coder Social home page Coder Social logo

gan_gengraph's Introduction

基于GAN的二次元头像生成模型

实验环境

python == 3.9.15
pytorch == 1.13.1
cuda == 11.7
CPU: 12th Gen Intel(R) Core(TM) i7-12700H
GPU: NVIDIA GeForce RTX 3070 Ti Laptop GPU

数据集下载

飞桨-动漫头像数据集

文件说明

config.py: 存放运行时的各项参数,可根据需要自行调整
model.py: 存放伸进网络结构的定义
train.py: 定义了神经网络的训练函数
generate.py: 定义了generate函数,可通过调用实现图像生成
main.py: 配置了简单的模型训练步骤

运行方式

进入主目录下运行一下命令即可

python main.py

若要调整迭代次数等参数在config.py文件中进行手动调整

gan_gengraph's People

Contributors

nirvana-apoc avatar

Stargazers

 avatar

Watchers

 avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.