This repo contains the code used to analyse the Cutting Tool Wear Dataset. You can find the dataset on Kaggle titled, "Cutting Tool Wear Audio Dataset".
The dataset is a collection of 1488 audio samples of End Mill Cutters of different levels of wear cutting through a piece of mild steel workpiece at two different RPMs using a Vertical Milling Machine. For more information about the dataset in particular, please refer to the Kaggle link. Please note, there was no addition of background noise to the dataset.
The code was tested with Python3.9 as the interpreter.
- Clone this repo.
- Install dependencies using the below command:
pip install -r requirements.txt- Download the dataset from Kaggle.
- Make changes to the
baseline.yamlfile related to the location of the dataset and neural network features. - run
baseline.pywith the following command:
python baseline.py- After completion, a
results.yamlwill be generated in theresultsdirectory and anROC Curve graphwill be generated inside themodeldirectory.
The code is heavily inspired by the MIMII_Baseline.
If this work renders helpful to you, please consider citing our paper:
@INPROCEEDINGS{10461855,
author={Soni, Nachiket and Kumar, Amit and Patel, Hardik},
booktitle={2023 IEEE 11th Region 10 Humanitarian Technology Conference (R10-HTC)},
title={Acoustic Analysis of Cutting Tool Vibrations of Machines for Anomaly Detection and Predictive Maintenance},
year={2023},
pages={43-46},
doi={10.1109/R10-HTC57504.2023.10461855}}