/stable-net

Unofficial pytorch implementation of StableNet

Primary LanguagePythonMIT LicenseMIT

StableNet

Unofficial implement of Deep Stable Learning for Out-Of-Distribution Generalization.

Example

Use the caltech101 dataset, And the saliency maps shows that StableNet does removes irrelevant feature information.

Model Saliency maps
resnet18 origin
resnet18_stable recon
resnet18 origin
resnet18_stable recon
resnet18 origin
resnet18_stable recon