IntelLabs / MART

Modular Adversarial Robustness Toolkit
BSD 3-Clause "New" or "Revised" License
16 stars 0 forks source link

General ImageVisualizerCallback #156

Open dxoigmn opened 1 year ago

dxoigmn commented 1 year ago

What does this PR do?

Writing images to disk a bad idea when we can write images to Tensorboard. This PR also genericizes PerturbedImageVisualizer into a configurable ImageVisualizer so that one can visualize perturbations, inputs, and perturbed inputs.

Type of change

Please check all relevant options.

Testing

Please describe the tests that you ran to verify your changes. Consider listing any relevant details of your test configuration.

Before submitting

Did you have fun?

Make sure you had fun coding 🙃