Advertisements

Deep Learning (DL) Framework Code

Hi Hi Hi,

Di postingan ini, saya mau share code yang saya bikin mengenai deep learning buat computer vision / image processing. Code deep learning yang saya buat berfokus untuk melakukan training end-to-end dari gambar ke gambar seperti yang tertera pada gambar berikut ini:

cnn

Framework yang saya buat terdiri dari empat file utama, yaitu:

  • main.py
  • dataset.py
  • frontend.py
  • network.py

Dalam mengembangkan framework ini, saya menggunakan library deep learning Keras sebagai wrapper dan TensorFlow sebagai backend. Selain itu, saya juga menggunakan OpenCV dan https://github.com/aleju/imgaug sebagai image augmentation library.  Detil mengenai masing-masing file adalah sebagai berikut:

Dataset.py

File ini berisi dua hal, yaitu data preparation dan data generator. Data preparation berfungsi untuk menyiapkan file path untuk setiap training data. Data generator berfungsi untuk membuat Keras sequence yang akan dipanggil pada saat melakukan training.

Network.py

File ini berisi master dan sub class dari network deep learning yang ingin diimplementasi. User dapat mengimplementasi semua network yang ingin diimplementasi pada file ini.

Frontend.py

File ini berfungsi sebagai file utama yang menghubungkan dataset.py dan network.py dalam sebuah kelas DLFramework. Fungsi train, predict, dan evaluate terdapat di dalam kelas DL Framework.

Main.py

File ini berisi fungsi-fungsi yang digunakan untuk memanggil kelas DLFramework.py

Code yang dibuat merupakan code untuk keperluan umum dan dapat diedit sesuai dengan keperluannya masing-maing. Pada code yang saya buat, saya memberi contoh dalam mengimplementasikan AOD Dehazing Network (http://openaccess.thecvf.com/content_iccv_2017/html/Li_AOD-Net_All-In-One_Dehazing_ICCV_2017_paper.html).

Code yang saya buat dapat dilihat dan didownload di link berikut ini: https://github.com/hengelpao/DL-Framework

Apabila ada komentar, silahkan memberikan komentar pada link github tersebut.

Terima kasih.

 

Ciao

 

Advertisements

Leave a Reply

Please log in using one of these methods to post your comment:

WordPress.com Logo

You are commenting using your WordPress.com account. Log Out /  Change )

Google photo

You are commenting using your Google account. Log Out /  Change )

Twitter picture

You are commenting using your Twitter account. Log Out /  Change )

Facebook photo

You are commenting using your Facebook account. Log Out /  Change )

Connecting to %s

%d bloggers like this: