c4dt / predictive-maintenance

An experiment to decentralize learning for predictive maintenance
0 stars 0 forks source link

Use the MVTech dataset #2

Closed ineiti closed 1 year ago

ineiti commented 1 year ago

Change the code in resnet-mlp/train.py to use the actual MVTech dataset for the training. Please make sure that the image themselves don't get committed to the github repo. Look at how I did it in the patchcore directory using .gitignore.