Demo framework for "The AI Mechanic: Acoustic Vehicle Characterization Neural Networks"
Requires Anaconda environment with numerous libraries installed: numpy, pytorch, librosa (and potentially more)
Also was tested using a NVIDIA 1080 Ti GPU, running via cpu not tested or recommended.
To run:
python demo.py --resume checkpoint/weights.pth
Please email [email protected] for link to trained model weights.
Framework is built upon the template from: https://github.com/victoresque/pytorch-template.