sheezashabbir / multitasklearning-multi-label-classification-using-cnn- Goto Github PK
View Code? Open in Web Editor NEWTraining a new classification network which should tell the identity label of an image and also be able to classify: ● Gender as Male/Female ● Wearing Glasses/ Not Wearing Glasses ● Smiling/ Not Smiling ● Wearing hat/ not wearing hat ● young/old. As we need 5 outputs from a single input. The custom CNN model consists of NN architecture with 5 heads return my the model. My model performance in Multilabel classification task is totally remarkable.