Skip to content

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

23 Commits
 
 
 
 

Repository files navigation

OpenMix

Codes for "OpenMix+: Revisiting Data Augmentation for Open Set Recognition". The corresponding paper can be found in [link].

  • Highlights

    1. Contrary to the view of "a good closed-set classifier is all you need for open set recognition" [link], we propose a new perspective that open set recognition requires the balance between classification and unknowns detection.

    2. We propose data augmentation strategies termed "OpenMix", which could promote the model's performance on unknowns detection.

  • Implimentation

    This repository provides a pytorch-version implementation of OpenMix which includes three data augmentation methods, i.e., OpenMixup, OpenCutout, OpenCutmix. The scripts are included in utils.py which can be directly called in your dataloader.

  • Notes

    Generally, OpenMixup is robust to foregrounds without rotational symmetry, and OpenCutout is robust to images with diverse backgrounds. Finetune the sample ratio of augmentation is helpful for achieving better AUROC results. In our experiments:

    For CIFAR10 and Imagenet, Known:OpenMixup:OpenCutout:OpenCutmix=1:4:1:1;

    For MNIST, Known:OpenMixup:OpenCutout:OpenCutmix=1:0:0:1 .

If you have any problems, feel free to contact me (jiangggss AT tju.edu.cn). Have fun and may it inspire your own idea :-)

About

Project for "OpenMix+: Revisiting Data Augmentation for Open Set Recognition"

Resources

Stars

0 stars

Watchers

1 watching

Forks

Releases

Packages

Contributors

Languages