This Repository is a deep learning pipeline for classification of a special noise in an assembly process. This Code is the base of our paper.
Our whole dataset is available at Google.drive. We labeled the data ourself and create 4325 data examples. If you clone our repository please set up the paths according to the path in our Jupter Notebooks.
Jypter Notebooks on Google Colab
- Select 'Runtime' -> 'Change runtime time' -> 'Python 3' (and 'GPU') before running the notebook.
- Ignore the warning: 'This notebook was not authored by Google.' Check 'Reset all runtimes before running' and click 'Run anyway'
Train and check out our CNN model at
or
Check out our XGB, SVM, RF, GNB Benchmarking
In the subfolder models you can access all our trained models and directly use them for your own predictions.