Otomatisasi Pewarnaan Citra Monokrom dengan Metode Generative Adversarial Network

Sistem pewarnaan otomatis pada citra greyscale diperlukan agar perbaikan pada citra monokrom dapat dilakukan secara cepat dan tanpa keahlian khusus. Dalam pengimplementasian sistem ini terdapat beberapa kendala seperti domain warna pada satu objek bisa sangat beragam, dimana setiap benda bisa memiliki lebih dari 1 kemungkinan warna. Pada penelitian ini penulis menggunakan SAGAN untuk melakukan pewarnaan otomatis pada dataset Hackaton Blossom (Flower Classification) yang bisa diakses pada situs resmi Kaggle, data ini berisikan citra 102 spesies bunga dengan jumlah data sebanyak 6500 data tes, 1000 data test dan 100 data validasi. SAGAN dipilih sebagai metode yang digunakan karena diklaim dapat menyelesaikan masalah image-to-image translation seperti style transfer, dan image colorization. Pada penelitian ini penulis berhasil mengimplementasikan dan melatih model SAGAN yang konvergen dengan memanfaatkan spectral normalization, selain itu penulis menambahkan SSIM loss untuk mengurangi munculnya artifact pada citra yang dihasilkan, model dilatih hingga 50 epoch dan berhasil memperoleh nilai PSNR diantara 51 dB hingga 73 dB dengan rata-rata nilai PNSR sebesar 61 dB.