Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Train fron scratch? #31

Open
danielsoy opened this issue Nov 14, 2022 · 2 comments
Open

Train fron scratch? #31

danielsoy opened this issue Nov 14, 2022 · 2 comments

Comments

@danielsoy
Copy link

Hey ppl!

Is it possible to train the model from scratch ?

How?
Thanks!

@antonemanuel
Copy link
Contributor

antonemanuel commented Nov 14, 2022

Hi!

Both padim and patchcore uses a pretrained backbone (e.g. resnet). You could use whatever backbone you want. For example you could train a resnet on your own data for some other task, like detection. But I'm not sure this would generate better results than using the pretrained nets in PyTorch (but it's possible).

The model fitting in this package however aims at generating a representation of the features to be classified from such a pretrained net. In the case of padim it creates a multivariate normal distribution of the features. And in patchcore it generates a core set of features.

@danielsoy
Copy link
Author

Thanks for the explanation related to the differences between patchcore and padim.

I just tried padim, will train with patchcore to see if the performance improves .

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants