Skip to content

StevenLiuWen/convlstm_anomaly_detection

 
 

Repository files navigation

#REMEMBERING HISTORY WITH CONVOLUTIONAL LSTM FOR ANOMALY DETECTION

By Weixin Luo$^{}$, Wen Liu$^{}$, Shenghua Gao

Introduction

All codes are based on Caffe. The main modifications are following

  • Adding a convolutional lstm layer
  • Changing the original Blob datastructure from 4-dimension tensor $(N \times C \times H \times W)$ to 5-dimension tensor $(T \times N \times C \times H \times W)$ tensor. Be careful that do not use cuDNN engien in all activation function layers in this motidified Caffe version, such as ReLU, sigmoid, tanh and so on, because cuDNN do not support 5-dimension in these activation function (but Conv layer is ok). If do that, it will be wrong.
  • Adding a video data layers

Requirements for compiling

Since all the code are implemented on Caffe, so that in order to compile successfully, it must satisfy all requirements of Caffe (see the installization instruction). Whatmore, it also need to compile with Opencv 3.+, because the video data layer uses opencv libraries.

Datasets

  • CUHK Avenue
  • UCSD Pedestrians 1 & 2
  • Subway Enter & Exit please to contact the orginal author to get the download links.

Training

As CUHK Avenue dataset for example, details are showd in the zstorm_conv_lstm_deconv/solver.prototxt and zstorm_conv_lstm_deconv/train.prototxt.

Testing or evaluation

Details are in the python shell, zstorm_conv_lstm_deconv/test.py.

License

All code are following the license of Caffe, and Caffe is released under the BSD 2-Clause license.

About

No description or website provided.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • C++ 79.9%
  • Python 9.6%
  • Cuda 5.8%
  • CMake 2.7%
  • MATLAB 0.9%
  • Makefile 0.7%
  • Shell 0.4%