Self-Supervised-Gans-Pytorch

用旋转自监督让 GAN 训练更稳,生成图像更逼真

Self-Supervised-Gans-Pytorch 是 CVPR 2019 论文《Self-Supervised GANs via Auxiliary Rotation Loss》的 PyTorch 实现,旨在解决传统 GAN 在无监督生成任务中训练不稳定、模式崩溃等问题。该项目通过引入辅助旋转损失(Auxiliary Rotation Loss)作为自监督信号,让判别器在区分真实与生成图像的同时,学习预测图像旋转角度,从而增强特征表示能力,提升生成图像的质量和多样性。核心能力包括:完整的训练流程、支持多种数据集(如 CIFAR-10、STL-10)、可配置的模型架构、以及基于旋转自监督的判别器设计。项目代码结构清晰,易于扩展和复现论文结果,适合研究自监督学习与 GAN 结合的研究者和开发者。

开源 free 图像生成
访问官网 ↗ GitHub ↗ 文档 ↗
GitHub 星标 ★ 66
维护状态 低维护
是否开源
定价模式 free

项目数据

分类图像生成
开发团队vandit15
所属国家
官网地址
定价模式free
价格说明开源项目,MIT许可证,完全免费,可自行部署使用。
访问状态
是否开源
开源协议MIT
主要语言Python
技术栈/模型computer-vision,cvpr,cvpr2019,deep-learning,gan,image-generation,machine-learning,pytorch,unsupervised-learning
GitHub 星标★ 66
30天Star增速
HF 下载量
上线时间2019-11-10 00:00:00
最近更新2025-09-23 00:00:00
维护状态低维护
中文支持
访问方式
移动端支持
综合评分
收录时间2026-08-09
浏览次数2

使用教程

核心亮点

  • 直接复现 CVPR 2019 论文,提供完整训练代码和配置,开箱即用
  • 引入旋转损失作为辅助任务,显著提升判别器特征质量,缓解模式崩溃
  • 支持 CIFAR-10、STL-10 等常用数据集,易于替换自定义数据

不足之处

  • 项目维护较少,依赖旧版 PyTorch,可能需适配新环境
  • 文档仅覆盖基本训练,缺少详细的超参数调优指南

适用场景

  • 学术研究:复现论文实验,验证自监督 GAN 效果
  • 教学演示:理解旋转损失在 GAN 中的作用
  • 基线对比:作为无监督生成任务的基准模型

替代项目

Self-Supervised-GANs (TensorFlow 官方实现)、BigGAN-PyTorch、SAGAN

项目介绍

Self-Supervised-Gans-Pytorch 是模型训练领域的开源项目,由 vandit15 开发,2019 年首次发布。

它收录于模型训练分类,该分类目前共有 518 个项目,GitHub 星标数为 66。

项目已超过三个月没有代码更新,维护节奏明显放缓,最近一次代码更新于 2025-09-23。MIT许可证,完全免费,可自行部署使用。

它主要面向的使用场景是:学术研究:复现论文实验,验证自监督GAN效果。同类可对比的替代方案包括 Self-Supervised-GANs (TensorFlow 官方实现)、BigGAN-PyTorch、SAGAN。

上一篇:denoising_diffusion

下一篇:lora

同类项目推荐

Long-RL 开源

让强化学习轻松驾驭超长序列,训练更稳更快

Long-RL: Scaling RL to Long Sequences (NeurIPS 2025)

★ 729 2026-08-09
TPA 开源

把注意力复杂度从平方降到线性,长序列不再卡顿

[NeurIPS 2025 Spotlight] TPA: Tensor ProducT ATTenTion Transformer (https://arxiv.or···

★ 463 2026-08-09
minimind 开源

两小时造出你的小模型,练手入门不求人

Train a 64M-parameter LLM from scratch in just 2h!

★ 62178 2026-08-09
pruna 开源

一键优化 AI 模型,推理提速又省资源。

Pruna is a model optimization framework built for developers, enabling you to delive···

★ 1286 2026-08-13