search

marco-rudolph / differnet

This is the official repository to the WACV 2021 paper "Same Same But DifferNet: Semi-Supervised Defect Detection with Normalizing Flows" by Marco Rudolph, Bastian Wandt and Bodo Rosenhahn.
215 stars 67 forks source link

Can this work process one channel pics? #36

Closed ghost closed 2 years ago

ghost commented 2 years ago

Hi, Thank you very much for your work. Can it be possible for this code to process only one channel of images?

marco-rudolph commented 2 years ago

Hi, since the feature extractor is pretrained on the RGB dataset ImageNet, you will have to convert your image into RGB. For converted grayscale images helpful features should still be found from my experience. Otherwise you would have to use a feature extractor that has been pretrained on your type of image.

Best regards Marco