Padim Training with large dataset

OpenAOI / anodet

Anomaly detection on images using features from pretrained neural networks.

MIT License

69 stars 22 forks source link

Padim Training with large dataset #10

Open AntonMaxen opened 3 years ago

AntonMaxen commented 3 years ago

Application gets out of memory while training, figure out a way to split training in stages, so application can be run on machines with moderate amount of memory resources. HDF5 filesystem could be solution to this. Functions:

extractEmbeddingVector
extractEmbeddingVectorsDataloader
getMeanCov