3,702 repos GH 3,587 / HF 115 · 11 categories · GitHub上のAI関連OSSを日本語で早見 · 毎日自動更新

DALLE-pytorch

★ 5.6k ⑂ 642 MIT 更新: 2024-02-17 GitHubで見る →

概要

このリポジトリは、OpenAIのDALL-E(テキストから画像を生成するTransformerモデル)をPyTorchで実装・再現したものです。

生成された画像の品質をランキングするためのCLIPも組み込まれています。

機械学習の研究者や開発者、特にPyTorchを使用して画像生成モデルやテキスト・画像間の関連性を探求したいユーザーを対象としています。

Colabでのトレーニングや推論の例も提供されており、さまざまなデータセットでの学習事例が紹介されています。

これにより、テキスト記述に基づいた画像生成モデルの構築と実験が可能になります。

互換性・特徴

  • Python
  • GPU必須
  • CLI
  • Colab対応
  • Text to Image
  • Deep Learning

基本情報

ライセンスMIT
Stars5,628
Forks642
カテゴリ画像生成
アクティビティmid

最新のissue

最新リリース: 1.6.6 (2023-05-24)

GitHub: https://github.com/lucidrains/DALLE-pytorch