HyperGAN/HyperGAN

Composable GAN framework with api and user interface

何を解決するか

HyperGAN は PyTorch を使用して、生成的対抗ネットワーク(GAN)の構築、トレーニング、共有を簡素化します。開発者、研究者、アーティストが手動でのネットワークアーキテクチャ設計やデータ準備の複雑さから解放される、組み合わせ可能なフレームワークを提供します。

どう動くか

このプロジェクトは、JSON 設定テンプレートに基づいてモデルを作成できる高レベルな API とコマンドラインインターフェース(CLI)を提供します。フォルダ内の画像(不完全なデータを含む)からデータを読み込み、並列トレーニングの高速化バックエンドをサポートします。モデルは ONNX 形式にエクスポートでき、他のプロジェクトで利用可能です。

対象ユーザー

PyTorch のコードをゼロから大量に書く必要なく、合成データや画像を生成したい開発者、AI 研究者、デジタルアーティスト。

特徴

  • 組み合わせ可能なアーキテクチャ:JSON でカスタムアーキテクチャを定義するか、カスタム PyTorch モジュールでコンポーネントを置き換える。
  • 柔軟なトレーニング:教師なし学習、転移学習、オンライン学習をサポート。
  • クロスプラットフォーム:Windows、OSX、Linux で動作。
  • データの耐障害性:不完全なデータセットを扱うために、自動的に画像のリサイズとクロッピングを処理。
  • モデル共有:異なる環境間でトレーニング済みモデルを共有・利用するための組み込みサポート。

関連

  • プロジェクト
  • プロジェクト
  • プロジェクト
  • プロジェクト