概要
本ツールは、PyTorchフレームワーク上で動作する画像から画像への変換(Image-to-Image Translation)を実現するライブラリです。
CycleGANによる非ペア画像変換とpix2pixによるペア画像変換の二つの主要な手法を提供し、様々な視覚タスク(スタイル変換、セマンティックセグメンテーションなど)に適用できます。
最新版はPython 3.11およびPyTorch 2.4+をサポートし、単一マシンでの複数GPUトレーニング(DDP)にも対応。
高速な推論を可能にするimg2img-turboや、効率的な学習を実現するCUTモデルにも言及しており、研究開発のニーズに応えます。
主に画像処理の研究者や開発者、特にPyTorchユーザーがGPU環境で利用することを想定しています。
互換性・特徴
- PyTorch
- Python
- GPU必須
- CLI
- 研究用途
基本情報
| ライセンス | NOASSERTION |
| Stars | 25,157 |
| Forks | 6,568 |
| カテゴリ | 画像生成 |
| アクティビティ | mid |
最新のissue
- 標準評価指標(MAE, MSE, PSNR, SSIM, FIDなど)の追加 (更新: 2026-04-16 / Add standard evaluation metrics (MAE, MSE, PSNR, SSIM, FID, etc.))
- 学習時の画像変換効果は良好だが、学習済み重みでの予測では著しく悪化する (更新: 2026-03-01 / The conversion effect between images is good during training, but it is much worse when predicting with the trained weights)
- masterブランチでマルチGPUケースの実行に失敗する (更新: 2025-12-18 / Fail to run the multi-gpu case in the master branch)
GitHub: https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix
