Giters

deel-ai / oodeel

Simple, compact, and hackable post-hoc deep OOD detection for already trained tensorflow or pytorch image classifiers.

Home Page:https://deel-ai.github.io/oodeel/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Add option to return features + labels in feature extractors

y-prudent opened this issue · comments

commented

Feature Extractors are lacking an option to return the labels in addition to the features in predict call. This would for instance benefit to mahalanobis implementation that has to iterate on the dataset in a for loop to get the labels (https://github.com/deel-ai/oodeel/blob/master/oodeel/methods/mahalanobis.py#L60C1-L74C41).