Code for constructing datasets

[ZuowenWang0000]

Hello, Could you please also release the code for constructing the robust and non-robust datasets? Specifically the code for C.4 section and C.5 section in the appendix of the paper.
Thanks!

[yw981]

+1
Thx

[wushan0125]

+1
Thx

[zhiyugege]

+1
Thx

https://github.com/MadryLab/robust_representations/blob/master/image_inversion.ipynb [for constructing robust datasets]

[ahadda5]

+1
Thx

https://github.com/MadryLab/robust_representations/blob/master/image_inversion.ipynb [for constructing robust datasets]

Not sure that is the correct code requested.
This notebook just applies a custom loss, scales the images for inversion and applies the adv-trained model to generate the inversion. What is important are the last two steps (argmin and Union till the whole Robust set is constructed),

[ahadda5]

Nevermind, it is answered here