Commit 367750e4 authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): fix the bug in plot

parent 0ecef949
Loading
Loading
Loading
Loading
+2 −4
Original line number Diff line number Diff line
@@ -9,10 +9,8 @@ sample_dir = 'sample'
if __name__ == '__main__':
    image = load_image(os.path.join(sample_dir, 'adversarial_input.png'))
    image_plot(
        [
        (image, 'Adversarial Noised'),
            (remove_adversarial_noise(image), 'Cleaned')
        ],
        (remove_adversarial_noise(image), 'Cleaned'),
        columns=2,
        figsize=(10, 6),
    )