halsayxi / ACGAN-Mnist

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

ACGAN-Mnist

文件说明

  • aigc_mn.py:实现接口类AiGcMn,类的初始化函数中完成模型加载等初始化工作。

    接口类AiGcMn提供一个接口函数generate,该函数的参数是一个整数型n维tensor(n是batch的大小,每个整数在0~9范围内,代表需要生成的数字),输出是n*1*28*28的tensor(n是batch的大小,每个1*28*28的tensor表示随机生成的数字图像),支持在CPU和GPU上运行随机产生输出图像。

    若直接运行aigc_mn.py,则会读取模型后随机生成10个不同的数字,将其分别保存在./results文件夹中。

  • models:模型文件夹,存有训练好的模型。

  • results:结果文件夹,保存生成的图片。

  • train:训练部分,保存用于训练模型的代码与数据。

    运行其中main.py即可进行训练,默认进行25轮训练,训练结果会产生图片保存在./train/results中,产生的模型保存在./train/models文件夹中。

  • report:报告。

About


Languages

Language:Python 100.0%