Skip to content

halak0013/new_esrgan

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

New ESRGAN ile Doğa Görüntüleri Netleştirme

Türkçe | English

Bu proje, derin öğrenme tabanlı ESRGAN (Enhanced Super-Resolution Generative Adversarial Network) modeli kullanılarak düşük çözünürlüklü doğa görüntülerinin kalite artırımı üzerine gerçekleştirilmiştir.

📌 Proje Tanımı

Görüntü süper çözünürlük (Super-Resolution), düşük çözünürlüklü (LR) bir görüntüden yüksek çözünürlüklü (HR) bir görüntü üretmeyi hedefler. Bu projede ESRGAN mimarisi ile bulanıklaştırılmış ya da kalite düşürülmüş doğa manzaraları netleştirilmiştir.


🖼️ Örnek Çıktılar

Aşağıda modelin düşük çözünürlüklü girişlerden ürettiği yüksek çözünürlüklü çıktılara ait örnekler yer almaktadır:

Test Görüntüsü 1

k1

Test Görüntüsü 2

k2

Test Görüntüsü 3

k3

TensorBoard Çıktısı

ssim_psnr_vgg


🧠 Kullanılan Model: ESRGAN

  • Generator: Residual-in-Residual Dense Block (RRDB) mimarisi ile detayların korunması ve geliştirilmesi.
  • Discriminator: Relativistic GAN yaklaşımı ile daha gerçekçi görüntüler üretme.
  • Kayıp Fonksiyonları:
    • VGG Perceptual Loss
    • Adversarial Loss
    • L1 Loss

📁 Veri Kümesi

⚙️ Eğitim Detayları

  • Ortam: Yerel GPU + Çevrimiçi GPU
  • Epoch: 25
  • Batch Size: 32
  • Optimizer: Adam
  • Scheduler: gan_custom, lambdaLR, ReduceLROnPlateau gibi farklı planlayıcılar test edilmiştir.
  • Model Kaydı: save_checkpoint() ve load_checkpoint() fonksiyonları ile yapılmıştır.

🔬 Deneysel Sonuçlar

Model PSNR SSIM
En iyi sonuç (Custom + Yeni Dataset) 43.03 0.9841

Tensorboard çıktıları, görsel karşılaştırmalar ve kayıpların detaylı analizi proje dosyalarında yer almaktadır.

🔧 Proje Yapısı


├── main.py
├── readme.md
├── src
│   ├── esrgan
│   │   ├── model.py
│   ├── test.py
│   ├── train.py
│   └── utils
│       ├── config.py
│       ├── data_loaders.py
│       ├── dataset_from_folder.py
│       ├── dataset.py
│       ├── loss.py
│       ├── scheduler_select.py
│       └── utils.py
└── test_images

🏁 Başlangıç

Projeyi çalıştırmak için:

python3 main.py train False

Test için:

python3 main.py test False

README dosyanıza çıktı görselleri için uygun bir bölüm başlığı aşağıdaki şekilde ekleyebilirsiniz:

👥 Katkıda Bulunanlar

  • Muhammet Halak
  • Abdullah Sina Korkmaz

📚 Kaynaklar


About

Image resolution enhancer with ESRGAN

Resources

License

Stars

Watchers

Forks

Contributors 2

  •  
  •