Skip to content

ShenZheng2000/img2img-turbo

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

97 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

🚀 Overview

We extend pix2pix-Turbo and CycleGAN-Turbo with a plug-and-play warp–unwarp strategy.

Specifically, this framework supports three tasks:

  • Human relighting
  • Driving scene relighting
  • Driving scene translation (weather & time-of-day)

⚙️ Installation

Setup and environment installation is in install.md.

📥 Dataset Preparation

Dataset download, preparation and warping instructions are in dataset.md.

⚡ Training and Testing

Model training and testing instructions are in train_test.md.

🔌 Warp–Unwarp Integration

Key warp–unwarp code insertions are marked with .

The main insertion points are:

  • Data loading → src/my_utils/training_utils.py (PairedDataset, UnpairedDataset)
  • Pix2Pix-Turbo → src/train_pix2pix_turbo.py
  • CycleGAN-Turbo → src/train_cyclegan_turbo.py
  • Warp utilities → src/warp_utils/

📌 TODO Lists

  • Add Gradio demo
  • Add arXiv link

🙏 Acknowledgement

This project builds upon the following excellent open-source works:

About

One-step image-to-image with Stable Diffusion turbo: sketch2image, day2night, and more

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages

  • Python 80.6%
  • Shell 18.4%
  • CSS 1.0%