Code for paper entitled "Improving Novelty Detection using the Reconstructions of Nearest Neighbours"

Last update: Dec 14, 2022

Related tags

Overview

`NLN`: Nearest-Latent-Neighbours

A repository containing the implementation of the paper entitled Improving Novelty Detection using the Reconstructions of Nearest Neighbours

Installation

Install conda environment by:

    conda create --name nln python=3.7

Run conda environment by:

    conda activate nln

Install dependancies by running:

    pip install -r dependancies

Additionally for training on a GPU run:

    conda install -c anaconda tensorflow-gpu=2.2.0

Replication of results in paper

Run the following to replicate the results for MNIST, CIFAR-10, Fashion-MNIST and MVTec-AD respectively

    sh experiments/run_mnist.sh
    sh experiments/run_cifar.sh
    sh experiments/run_fmnist.sh
    sh experiments/run_mvtec.sh

Or to execute all experiments sequentially the following script can be run:

    sh experiments/run_all.sh

MVTec-AD usage

You will need to download the MVTec anomaly detection dataset and specify the its path using -mvtec_path command line option.

Training

Run the following:

    python main.py -anomaly_class <0,1,2,3,4,5,6,7,8,9,bottle,cable,...> \
                   -percentage_anomaly <float> \
                   -limit <int> \
                   -epochs <int> \
                   -latent_dim <int> \
                   -data <MNIST,FASHION_MNIST,CIFAR10,MVTEC> \
                   -mvtec_path <str>\
                   -neighbors <int(s)> \
                   -algorithm <knn> \
		   -patches <True, False> \
		   -crop <True, False> \
		   -rotate <True, False> \
		   -patch_x <int> \    
		   -patch_y <int> \    
		   -patch_x_stride <int> \    
		   -patch_y_stride <int> \    
		   -crop_x <int> \    
		   -crop_y <int> \

Reporting Results

Run the following given the correctly generated results files:

    python report.py -data <MNIST,CIFAR10,FASHION_MNIST,MVTEC> -seed <filepath-seed>

Licensing

Source code of NLN is licensed under the MIT License.

Code for paper entitled "Improving Novelty Detection using the Reconstructions of Nearest Neighbours"

Related tags

Overview

`NLN`: Nearest-Latent-Neighbours

Installation

Replication of results in paper

MVTec-AD usage

Training

Reporting Results

Licensing

Owner

Michael (Misha) Mesarcik

Code for the SIGGRAPH 2022 paper "DeltaConv: Anisotropic Operators for Geometric Deep Learning on Point Clouds."

The code is for the paper "A Self-Distillation Embedded Supervised Affinity Attention Model for Few-Shot Segmentation"

S2-BNN: Bridging the Gap Between Self-Supervised Real and 1-bit Neural Networks via Guided Distribution Calibration (CVPR 2021)

An Ensemble of CNN (Python 3.5.1 Tensorflow 1.3 numpy 1.13)

Spatially-Adaptive Pixelwise Networks for Fast Image Translation, CVPR 2021

This is the repository for paper NEEDLE: Towards Non-invertible Backdoor Attack to Deep Learning Models.

Anti-UAV base on PaddleDetection

Emotion classification of online comments based on RNN

Styled text-to-drawing synthesis method. Featured at the 2021 NeurIPS Workshop on Machine Learning for Creativity and Design

A PyTorch implementation of "Semi-Supervised Graph Classification: A Hierarchical Graph Perspective" (WWW 2019)

Weakly Supervised Learning of Rigid 3D Scene Flow

Colossal-AI: A Unified Deep Learning System for Large-Scale Parallel Training

A python implementation of Deep-Image-Analogy based on pytorch.

VIMPAC: Video Pre-Training via Masked Token Prediction and Contrastive Learning

Official Implementation of "DialogLM: Pre-trained Model for Long Dialogue Understanding and Summarization."

f-BRS: Rethinking Backpropagating Refinement for Interactive Segmentation

A deep learning framework for historical document image analysis

smc.covid is an R package related to the paper A sequential Monte Carlo approach to estimate a time varying reproduction number in infectious disease models: the COVID-19 case by Storvik et al

First-Order Probabilistic Programming Language

Efficient Training of Audio Transformers with Patchout

Code for paper entitled "Improving Novelty Detection using the Reconstructions of Nearest Neighbours"

Related tags

Overview

NLN: Nearest-Latent-Neighbours

Installation

Replication of results in paper

MVTec-AD usage

Training

Reporting Results

Licensing

Owner

Michael (Misha) Mesarcik

Code for the SIGGRAPH 2022 paper "DeltaConv: Anisotropic Operators for Geometric Deep Learning on Point Clouds."

The code is for the paper "A Self-Distillation Embedded Supervised Affinity Attention Model for Few-Shot Segmentation"

S2-BNN: Bridging the Gap Between Self-Supervised Real and 1-bit Neural Networks via Guided Distribution Calibration (CVPR 2021)

An Ensemble of CNN (Python 3.5.1 Tensorflow 1.3 numpy 1.13)

Spatially-Adaptive Pixelwise Networks for Fast Image Translation, CVPR 2021

This is the repository for paper NEEDLE: Towards Non-invertible Backdoor Attack to Deep Learning Models.

Anti-UAV base on PaddleDetection

Emotion classification of online comments based on RNN

Styled text-to-drawing synthesis method. Featured at the 2021 NeurIPS Workshop on Machine Learning for Creativity and Design

A PyTorch implementation of "Semi-Supervised Graph Classification: A Hierarchical Graph Perspective" (WWW 2019)

Weakly Supervised Learning of Rigid 3D Scene Flow

Colossal-AI: A Unified Deep Learning System for Large-Scale Parallel Training

A python implementation of Deep-Image-Analogy based on pytorch.

VIMPAC: Video Pre-Training via Masked Token Prediction and Contrastive Learning

Official Implementation of "DialogLM: Pre-trained Model for Long Dialogue Understanding and Summarization."

f-BRS: Rethinking Backpropagating Refinement for Interactive Segmentation

A deep learning framework for historical document image analysis

smc.covid is an R package related to the paper A sequential Monte Carlo approach to estimate a time varying reproduction number in infectious disease models: the COVID-19 case by Storvik et al

First-Order Probabilistic Programming Language

Efficient Training of Audio Transformers with Patchout

`NLN`: Nearest-Latent-Neighbours