Hi @mtailanian, first of all, congrats on your submission, hope you'll get the paper published.

If you would like to contribute, we would be happy to see U-Flow implemented in anomalib. Thanks!

Thanks @samet-akcay ! I'll start working on this!

Hello again @samet-akcay
I already implemented the basic functionalities, but I have a doubt about how to include the segmentation method.

Let me add some context: As a final stage of U-Flow, we propose a method for segmenting the anomalies, using an a contrario approach. Basically, it performs a series of statistical tests and decides for each pixel if it is anomalous or not, taking as input the output of the Normalizing Flow graph.
This part of the method is slower than the rest, and should only be called if a segmentation mask is required.

I'm trying to understand what's the best way of including it, but I'd like to discuss it to get some feedback from this community!

(I can start a PR if you think that looking at the code may help)

Thanks in advance!

Since U-Flow has been implemented, I'm now closing this discussion.