Progress of alpha matting project
Project aims at generating a large training dataset for alpha matting, training the model and converting the model to ONNX to
be used with OpenCV's DNN module and to integrate some of the best computer vision based best alpha matting algorithms into
OpenCV.
-
A large dataset which can be used to train models of alpha matting.
-
Implement the paper: "AlphaGAN Matting" by Sebastian Lultz.
-
Trained model by the use of the generated dataset
-
Convert model to ONNX and provide a running example in OpenCV's DNN module
-
Implement the paper: "A Global Sampling Method for Alpha Matting" by Kaiming He et al. Implement the paper: "Designing Effective Inter-Pixel Information Flow for Natural Image Matting" by Yagiz et. al."
-
Experiments comparing results to existing alpha matting algorithms at alphamatting.com
alphamatting.com Comparison of many methods, datasets etc
Excellent C++ and Python coding skills, Experience in training deep learning models in Tensorflow/Pytorch, Experience in
Computer Vision, Experience in Deep Learning
Mentors: Steven Puttemans,Gholamreza Amayeh,Sunita Nayak
Difficulty: Medium-Hard
-
Read and understood the architecture of each paper. Wrote an article explaining the papers Understanding AlphaGAN matting, Paper Summary:Global sampling method for Alpha matting
-
Learnt how to use gimp, to create the dataset for the deep learning based model.
-
Implemented Global sampling based method for alpha matting.
-
Verified the implementation is correct. Checked the implementation on Alphamatting.com. Here are the results.Link (Global sampling method-Vedanta is my implementation)
-
Implemented AlphaGAN matting, but the results even when trained on Adobe dataset is not satisfactory.
-
Generated alpha matte for about 100 images using gimp.Link
-
Generated alpha matte for about 250 images using gimp.[Link](](https://drive.google.com/drive/folders/12esYUgcR5PDl_wfZXF9dBpNwvO_60yL9)
-
Implemented muliple architectures for AlphaGAN matting. The results are much better than before but not as good as the original implementation.Following architectures have been implemented:
a. Generator with Resnet50 Encoder
b. Generator with Deeplabv3 encoder