MEMORABLE
This repository contains the source code accompanying our ICCV 2021 paper.
A Machine Teaching Framework for Scalable Recognition
Pei Wang, Nuno Vasconcelos.
In ICCV, 2021.
@InProceedings{wang2021gradient,
author = {Wang, Pei and Vasconcelos, Nuno},
title = {A Machine Teaching Framework for Scalable Recognition},
booktitle = {Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV)},
month = {October},
year = {2021}
}
Requirements
- The project was implemented and tested in Python 3.5 and Pytorch 1.0. Other versions should work after minor modification.
- NVIDIA GPU and cuDNN are required to have fast speeds. For now, CUDA 8.0 with cuDNN 6.0.20 has been tested. The other versions should be working.
Datasets
Butterflies and Chinese Characters, Gull are used. Please organize them as below after download,
datasets
|_ butterflies_crop
|_ images
|_ Viceroy
|_ ...
|_ chinese_chars
|_ images
|_ grass
|_ ...
|_ CUBgull
|_ images
|_ CaliforniaGull
|_ ...
Implementation details
To generate counterfactual explanations
get_all_CE_butterflies.py
get_all_CE_gull.py
by SimCLR model
get_all_CE_butterflies_simclr.py
get_all_CE_gull_simclr.py
To generate teaching images and explanations
train_butterflies_CMaxGrad.py
train_gull_CMaxGrad.py
by SimCLR model
train_butterflies_CMaxGrad_simclr.py
train_gull_CMaxGrad_simclr.py
For the SimCLR, we used repo for training. Thanks for their open-source. For standard model, we trained by
train_butterflies_resnet18.py
train_gull_resnet18_linear.py
For questions, feel free to reach out
Pei Wang: [email protected]
26 Nov 12, 2021
15 Dec 15, 2022
1.8k Jan 04, 2023
76 Dec 11, 2022
4 Dec 28, 2021
18 Sep 16, 2022
32 Nov 13, 2022
3 Oct 28, 2022
1 Nov 18, 2021
70 Oct 29, 2022
38 Dec 12, 2022
49 Dec 20, 2022
15 Dec 13, 2022
23 Dec 16, 2022
165 Oct 25, 2022
89 Dec 12, 2022
32 Dec 25, 2022
110 Dec 27, 2022
11 Nov 15, 2022
44 Dec 03, 2022