Utilizing PITI for Generating Autonomous UAV Images in Natural Environments.

Competition Name: Generative-AI Navigation Information Competition for UAV Reconnaissance in Natural Environments I：Image Data Generation
Team Name: TEAM_5101
Team Members: @Tsao666, Me
Final Competition Results:

Testing Dataset FID Score Rank

Private 89.09644 4

Public 88.878136 6
Competition Report
Key Technologies:
1. PITI (By Me)
2. DP_GAN (By @Tsao666)
3. Ensemble (By @Tsao666 and Me)

Testing Dataset	FID Score	Rank
Private	89.09644	4
Public	88.878136	6

Pretraining is All You Need for Image-to-Image Translation

Tengfei Wang, Ting Zhang, Bo Zhang, Hao Ouyang, Dong Chen, Qifeng Chen, Fang Wen
2022

paper | project website | video | online demo

Introduction

We present a simple and universal framework that brings the power of the pretraining to various image-to-image translation tasks.

Diverse samples synthesized by our approach.

Set up

Installation

git clone https://github.com/PITI-Synthesis/PITI.git
cd PITI

Environment

sudo apt-get update
sudo apt-get install openmpi-bin libopenmpi-dev -y
conda env create -f environment.yml
conda activate PITI
conda install -c conda-forge openmpi -y
pip install mpi4py==3.0.3 dlib==19.22.1
pip install gradio

Pretrained Models

Please download pre-trained models for both Base model and Upsample model, and put them in ./ckpt.

Model	Task	Dataset
Base-64x64	Mask-to-Image	Trained on COCO.
Upsample-64-256	Mask-to-Image	Trained on COCO.
Base-64x64	Sketch-to-Image	Trained on COCO.
Upsample-64-256	Sketch-to-Image	Trained on COCO.

If you fail to access to these links, you may alternatively find our pretrained models here.

Training

Preparation

Download the following pretrained models into ./ckpt/.

Model	Task	Dataset
Base-64x64	Mask-to-Image	Trained on COCO.
Upsample-64-256	Mask-to-Image	Trained on COCO.

Preprocess

Run the notebook preprocess.ipynb to preprocess training dataset.

Start Training

Taking mask-to-image synthesis as an example: (sketch-to-image is the same)

Finetune the Base Model

Modify mask_finetune_base.sh and run:

bash mask_finetune_base.sh

Inference

Run the notebook generate-example.ipynb to generate output images.

Citation

If you find this work useful for your research, please cite:

@misc{
  title  = {Utilizing PITI for Generating Autonomous UAV Images in Natural Environments},
  author = {Zhe-Yu Guo},
  url    = {https://github.com/Tianming8585/PITI},
  year   = {2024}
}

Acknowledgement & References

Thanks for PITI for sharing their code and pretrained models.

PITI-Synthesis/PITI

@article{wang2022pretraining,
 title = {Pretraining is All You Need for Image-to-Image Translation},
  author = {Wang, Tengfei and Zhang, Ting and Zhang, Bo and Ouyang, Hao and Chen, Dong and Chen, Qifeng and Wen, Fang},
  journal={arXiv:2205.12952},
  year = {2022},
}

Tsao666/DP_GAN

@misc{
  title  = {Apply DP-GAN on Generative-AI Navigation Information Competition for UAV Reconnaissance in Natural Environments},
  author = {Wei-Chun Tsao},
  url    = {https://github.com/Tsao666/DP_GAN},
  year   = {2024}
}

Tianming8585 / PITI