CNN-based screw classifier
End-to-end deep learning pipeline for screw identification from CAD renders and real video frames, served through a Telegram bot.
- Role
- Led the pipeline, team of 4
- Where
- Nanyang Technological University
- When
- Jan 2022 – Apr 2022
- Stack
- Python, VGG16, EfficientNet, InceptionV3
What I did
- Led an end-to-end deep learning pipeline in a team of 4 for screw identification, combining simulated CAD data, real-world video-derived images, augmentation, and transfer learning.
- Evaluated VGG16, EfficientNet, and InceptionV3, and integrated the final classifier with a Telegram-based inference interface.
Outcome
- 82% test accuracy.