keras-team / keras-cv

Industry-strength Computer Vision workflows with Keras
Other
1k stars 333 forks source link

Anomaly detection models (vision) #532

Closed innat closed 1 year ago

innat commented 2 years ago

Short Description

164736543-0ef58c66-dfb0-47c5-8e6e-26774dbc6fd3

Anomaly detection is an important category of common problem mostly in manufacturing industry. There is a good amount of ML models (and active research) to automate the anomaly detection task. I think it would be good addition to add these models in keras_cv.models.

Papers

Benchmark dataset

Metrics

Existing Implementations

Currently, there are not much strong support available for anomaly detection (vision) in tensorflow.keras. In contrast, a pytorch-lighting based implementations, named anomalib are available, maintained by a OpenVINO team. Currently it offers 10 different anomaly detection (vision) models and actively developing. These model are able to run in supervised (with binary mask) or unsupervised manner (with no mask).

Other Information

If these models are welcomed, we can close this issue and create new ones for these models each.

bhack commented 2 years ago

Some of these models are available in a TF/Keras flavour at https://github.com/Burf/tfdetection

/cc @burf If he Is interested to contribute.

bhack commented 2 years ago

Is MVTec saturated? https://paperswithcode.com/sota/anomaly-detection-on-mvtec-ad

innat commented 2 years ago

Yeah, looks so. Models like patch-core, or fast-flow (transformer based) left less room for improvement on this dataset.

LukeWood commented 1 year ago

At this time we're focusing on ironing out issues in our current offerings before expanding too broadly, lets revisit anomaly detection in the future when the more standard workflows are solved.