Skip to content

ristea/sspcab

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

15 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Self-Supervised Predictive Convolutional Attentive Block for Anomaly Detection - CVPR 2022 Oral (official repository)

We propose to integrate the reconstruction-based functionality into a novel self-supervised predictive architectural building block. The proposed self-supervised block is generic and can easily be incorporated into various state-of-the-art anomaly detection methods.

The open-access paper can be found at: https://arxiv.org/pdf/2111.09099.pdf

This code is released under the CC BY-NC-SA 4.0 license.


map


Information

Our kernel is illustrated in the picture below. The visible area of the receptive field is denoted by the regions Ki, ∀i ∈ {1, 2, 3, 4}, while the masked area is denoted by M. A dilation factor d controls the local or global nature of the visible information with respect to M. map

Implementation

We provide implementation for both PyTorch and Tensorflow in the sspcab_torch.py and sspcab_tf.py scripts.

In order to work properly, you need to have a python version newer than 3.6 (we used the python 3.6.8 version).

Citation

If you use our block in your own work, please don't forget to cite us:

@inproceedings{Ristea-CVPR-2022,
  title={Self-Supervised Predictive Convolutional Attentive Block for Anomaly Detection},
  author={Ristea, Nicolae-Catalin and Madan, Neelu and Ionescu, Radu Tudor and Nasrollahi, Kamal and Khan, Fahad Shahbaz and Moeslund, Thomas B and Shah, Mubarak},
  booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},
  year={2022}
}

Feedback

You can send your questions or suggestions to:

r.catalin196@yahoo.ro, raducu.ionescu@gmail.com

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages