Skip to content
Snippets Groups Projects
Devashish Lohani's avatar
Devashish Lohani authored
c9bd72b5
Name Last commit Last update
README.md
Testing.py
Train.py

memAE

This is an official implementation of paper "Unsupervised and Adaptive Perimeter Intrusion Detector", at ICIP 2022.

Requirements

  • PyTorch == 1.4.0

  • Python==3.7.6

  • ./requirement.sh

Prepare dataset

prepare_data.sh dataset datapath
dataset: i_LIDS
datapath: the path that you want to save the data, i.e., ./dataset/i_LIDS/

Train the model

Run python Train.py with parameters like dataset, frame length, width, etc.

Evaluate the model

python Testing.py
dataset: i_LIDS
datapath: the path that you saved the data
version: experiment version
ckpt: the checkpoint step
expdir: the path that you saved the model checkpoint

Note: The code will be cleaned and commented soon.