Textile Defect Detection with Variational Autoencoder and Gaussian Mixture Model. PLEASE NOTE: The implementation is not complete.
AITEX: Download from AITEX website. For automatic download:
cd utils/
./get_aitex.sh
cd ..
To train the model, use:
python train_network.py
To evaluate the model:
python test_network.py
Python 3.9 with PyTorch 1.9.0. Use the file environment.yml
for the conda environment.