https://github.com/mujadded/boltvision-vit
BoltVision
https://github.com/mujadded/boltvision-vit
Last synced: 18 days ago
JSON representation
BoltVision
- Host: GitHub
- URL: https://github.com/mujadded/boltvision-vit
- Owner: Mujadded
- Created: 2024-03-08T09:34:05.000Z (over 2 years ago)
- Default Branch: main
- Last Pushed: 2024-03-08T10:11:18.000Z (over 2 years ago)
- Last Synced: 2025-11-13T22:25:12.966Z (8 months ago)
- Language: Jupyter Notebook
- Size: 8.66 MB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# BoltVision: High Accuracy Classification of Missing Bolts in Train Components
## Introduction
BoltVision is a comparative analysis of various machine learning models including Convolutional Neural Networks (CNNs), Vision Transformers (ViTs), and Compact Convolutional Transformers (CCTs), aimed at accurately classifying missing bolts in train components. This repository contains the code and models used in the study published in Machines (MDPI). For more information, read our paper: [BoltVision: A Comparative Analysis of CNN, CCT, and ViT](https://www.mdpi.com/2075-1702/12/2/93).
## Setup
We would need pytorch mainly for this, we also have some model with tensorflow
## Usage
Different models are in different files to test properly. The file names are a good indication.
## Data
Not including the dataset and university has the rights to it
## Results
This study illustrates the superior assessment capabilities of these models and discusses their effectiveness in addressing the prevalent issue of edge devices. Results show that BoltVision, utilising a pre-trained ViT base, achieves a remarkable 93% accuracy in classifying missing bolts.
## Citation
If you use BoltVision in your research, please cite our paper:
```
@article{alif2024boltvision,
title={BoltVision: A Comparative Analysis of CNN, CCT, and ViT in Achieving High Accuracy for Missing Bolt Classification in Train Components},
author={Alif, Mujadded Al Rabbani and Hussain, Muhammad and Tucker, Gareth and Iwnicki, Simon},
journal={Machines},
volume={12},
number={2},
pages={93},
year={2024},
publisher={MDPI}
}
```
## Contact
mujadded.alif@gmail.com