Skip to content

Latest commit

 

History

History
3 lines (3 loc) · 572 Bytes

README.md

File metadata and controls

3 lines (3 loc) · 572 Bytes

LuggageClassification

Computer vision project to predict number of wheels of luggage This logic ensemble pretrained models Inception, ResNet and Densenet to classify images. Two FC layers are added on top of pretrained model output. The transfer learning phase fixed the pretrained model and train only ensembling FCs. To fasten up the training speed, data loaders are created which will have output of pretrained networks. Hence, in training, images are embedded using these models once and used it to finetune the FC. This reduces training time to minutes from hours.