没有合适的资源?快使用搜索试试~ 我知道了~
温馨提示
TinyGAN BigGAN; 知识提炼; 黑盒子; 快速培训; 16倍压缩 该存储库包含以下论文的官方PyTorch实现: TinyGAN:提取BigGAN进行条件图像生成(ACCV 2020) 张婷云和陆志仁 摘要:生成对抗网络(GAN)已成为生成图像建模的强大方法。 但是,GAN的训练不稳定(尤其是在大规模,复杂的数据集上)是众所周知的。 尽管BigGAN的最新工作显着提高了ImageNet上图像生成的质量,但它需要一个庞大的模型,这使其很难在资源受限的设备上进行部署。 为了减小模型的大小,我们提出了一种用于压缩GAN的黑匣子知识蒸馏框架,该框架强调了稳定而有效的训练过程。 以BigGAN作为教师网络,我们设法训练一个更小的学生网络来模仿其功能,从而在Inception和FID分数上实现有竞争力的表现,但生成器的参数减少了16倍。 训练后的模型位于gan / models(
资源推荐
资源详情
资源评论
收起资源包目录
ACCV_TinyGAN-master.zip (29个子文件)
ACCV_TinyGAN-master
cls_weight_reduce.npy 500KB
noises.npy 4.88MB
requirements.txt 52B
Get_ImageNet_FID
stat_real_ani.npz 32.01MB
my_dump_real_mean_cov.py 4KB
my_dump_real_act.py 5KB
readme.txt 479B
train.sh 715B
gan
models
model_30.tar 73.02MB
model2_30.tar 73.02MB
intra_fid_scores_real_small.npy 2KB
inception_tf.py 3KB
demo.ipynb 1.17MB
Get_BigGAN_FID
intra_fid_tf.py 2KB
dump_bigGAN_acts.py 3KB
class2idx.pkl 20KB
utils.py 3KB
README.md 2KB
data_loader.py 6KB
main.py 7KB
misc
TinyGAN_flow.png 86KB
TinyGAN_demo.png 526KB
extract_big.py 2KB
plot_corr_big_tiny.py 584B
logger.py 419B
numpy_fid.py 2KB
model.py 10KB
eval.sh 720B
solver.py 20KB
共 29 条
- 1
资源评论
火器营松老三
- 粉丝: 20
- 资源: 4649
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功