Update deprecated torchvision - Githubissues

OpenAOI / anodet

Anomaly detection on images using features from pretrained neural networks.

MIT License

69 stars 22 forks source link

Update deprecated torchvision #34

Closed wlinds closed 11 months ago

wlinds commented 11 months ago

This fix addresses the deprecation warning found in feature_extraction.ResnetEmbeddingsExtractor

ResNet18_Weights.DEFAULT equals to ResNet18_Weights.IMAGENET1K_V1

Same with wide_resnet50.

https://pytorch.org/vision/main/models/generated/torchvision.models.resnet18.html

https://pytorch.org/vision/main/models/generated/torchvision.models.wide_resnet50_2.html

Additionally,

gitignore updated to ignore venv and DS_Store