search

xiahaifeng1995 / STPM-Anomaly-Detection-Localization-master

This is an unofficial implementation of the paper “Student-Teacher Feature Pyramid Matching for Unsupervised Anomaly Detection”.
MIT License
45 stars 6 forks source link

Pretrained weights ? #2

Open trungpham2606 opened 3 years ago

trungpham2606 commented 3 years ago

Dear @xiahaifeng1995 , Thank you for such a nice implementation, your result is extremely great. I want to test it but I dont see where the pretrained weight is ?

KYKong1997 commented 3 years ago

I think you don't need the pretrained weight. The pytorch library will download it automatically.

trungpham2606 commented 3 years ago

@KYKong1997 Pytorch will automatically download the pretrained model of resnet18. The weights for the backbone model, not the final weights after training. image Correct me if I missed something in the code.