Confident_classifier

by alinlab

Training Confidence-Calibrated Classifier for Detecting Out-of-Distribution Samples / ICLR 2018

138 Stars 37 Forks Last release: Not found MIT License 20 Commits 0 Releases

Available items

No Items, yet!

The developer of this repository has not created any items for sale yet. Need a bug fixed? Help with integration? A different license? Create a request here:

Training Confidence-Calibrated Classifier for Detecting Out-of-Distribution Samples

This project is for the paper "Training Confidence-Calibrated Classifier for Detecting Out-of-Distribution Samples". Some codes are from odin-pytorch.

Preliminaries

It is tested under Ubuntu Linux 16.04.1 and Python 2.7 environment, and requries Pytorch package to be installed:

  • Pytorch: Only GPU version is available.

Downloading Out-of-Distribtion Datasets

We use download links of two out-of-distributin datasets from odin-pytorch:

Training scripts

Test scripts

  • test.sh
    --dataset --outdataset --pretrainednet \ --dataset = name of in-distribution (svhn or cifar10) \ --outdataset = name of out-of-distribution (svhn, cifar10, lsun or imagenet) \ --pretrainednet = path to pretrainednet

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.