GlassyWing / stable-net

Unofficial pytorch implementation of StableNet

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About

Unofficial pytorch implementation of StableNet

License:MIT License


Languages

Language:Python 100.0%