oskarnatan / mtl-bisaai

Compact Semantic Segmentation and Depth Estimation with Multi-task Learning

Multi-task AI on Google Colab: Simultaneous Image Segmentation and Depth Estimation

Implementasi program untuk webinar BISA AI - Kampus Merdeka:

Run di Google Colab:

1. Login akun google masing-masing
2. Run google colab ini atau ini
   • Ganti VM resource ke GPU untuk pemrosesan yang lebih cepat
   • Ganti variable 'rootdir' sesuai dengan direktori project di google drive anda masing-masing

Run di Local PC:

1. Download dan extract dataset cityscapes
   • leftImg8bit_trainvaltest.zip (11GB)
   • gtFine_trainvaltest.zip (241MB)
   • disparity_trainvaltest.zip (3.5GB)
2. Run jupyter notebook dan buka mtl_seg_dep_local.ipynb
3. Jalankan program sesuai instruksi yang tertulis di notebook
   • Ganti variable 'rootdir' sesuai dengan direktori project di local PC anda masing-masing
4. Trained model untuk inference dapat didownload di sini