2404.03898

Total: 1

#1 VoltaVision: A Transfer Learning model for electronic component classification [PDF3] [Copy] [Kimi1] [REL]

Authors: Anas Mohammad Ishfaqul Muktadir Osmani, Taimur Rahman, Salekul Islam

In this paper, we analyze the effectiveness of transfer learning on classifying electronic components. Transfer learning reuses pre-trained models to save time and resources in building a robust classifier rather than learning from scratch. Our work introduces a lightweight CNN, coined as VoltaVision, and compares its performance against more complex models. We test the hypothesis that transferring knowledge from a similar task to our target domain yields better results than state-of-the-art models trained on general datasets. Our dataset and code for this work are available at https://github.com/AnasIshfaque/VoltaVision.

Subjects: Computer Vision and Pattern Recognition , Machine Learning

Publish: 2024-04-05 05:42:23 UTC