This project demonstrates realistic shadow generation on natural images using a custom-trained GAN model. Given a deshadowed input image, the model learns to cast shadows in appropriate regions, conditioned on the shadow mask and instance segmentation.
1_shadow_generation_code.ipynb: Main Colab notebook for model inferenceassets/: Folder containing input, output, and mask visualizationsmodels/: Generator and discriminator architecturesutils/: Image preprocessing and visualization functions
- Input: A deshadowed RGB image
- Preprocessing: Generates:
- Shadow mask (binary)
- Instance mask (semantic segmentation)
- GAN-based Generation: Produces a realistic shadow-cast image
- Post-processing: Visualization and export
- Python, PyTorch, NumPy, OpenCV
- Google Colab for experimentation
- GANs: U-Net Generator, PatchGAN Discriminator
- This is a research-oriented prototype for cast shadow synthesis.
- Ideal for use in training data augmentation, simulation, or visual effects.
Sample image of zebra sourced from open dataset used for testing.