wangdeyu / abnormal_environment_detection

Abnormal Environment Detection

Enviroment setup

To install requirements:

pip install -r requirements.txt

Dataset

Please download dataset here:

• Dataset

Pre-trained Weights

Please download pre-trained weights here:

• ResNet50 Weights

Train

To train convolutional neural network (embedding):

cd Code
python TRAIN_crossentropy.py # for training unsupervised setting
python TRAIN_crossentropy_finetune_with_supcon.py # for finetuning

Evaluation

To evaluate and visualize before finetuning:

cd Code
python VISUALIZE_crossentropy.py

Clustering Results before Finetuning.

To evaluate and visualize after finetuning:

cd Code
python VISUALIZE_crossentropy_finetune_with_supcon.py

Clustering Results after Finetuning.