search

deel-ai / oodeel

Simple, compact, and hackable post-hoc deep OOD detection for already trained tensorflow or pytorch image classifiers.
https://deel-ai.github.io/oodeel/
MIT License
52 stars 2 forks source link

feat: torch feature extractor impl #4

Closed Matt3164 closed 1 year ago

Matt3164 commented 1 year ago

Pytorch Feature Extractor implementation.

Matt3164 commented 1 year ago

Not ready yet but available for review.

TODO:

Matt3164 commented 1 year ago

Should we add pytorch in the requirements.txt?

My opinion is no since the default backend is Keras/Tensorflow.

Matt3164 commented 1 year ago

Ready for review and merge!

paulnovello commented 1 year ago

Thank you Yannick and Matthieu, ready for merge!