Merge pull request #60 from deepghs/dev/adversarial

*.pt

/runs

/YOLOv8

.tests

 No newline at end of file

imgutils.restore.adversarial

====================================

.. currentmodule:: imgutils.restore.adversarial

.. automodule:: imgutils.restore.adversarial

remove_adversarial_noise

-----------------------------------

.. autofunction:: remove_adversarial_noise

import random

from benchmark import BaseBenchmark, create_plot_cli

from imgutils.restore.adversarial import remove_adversarial_noise

class AdversarialRemovalBenchmark(BaseBenchmark):

    def load(self):

        pass

    def unload(self):

        pass

    def run(self):

        image_file = random.choice(self.all_images)

        _ = remove_adversarial_noise(image_file)

if __name__ == '__main__':

    create_plot_cli(

        [

            ('Adversarial Removal', AdversarialRemovalBenchmark()),

        ],

        title='Benchmark for Adversarial Removal Algorithm',

        run_times=5,

        try_times=10,

    )()

import os

from imgutils.data import load_image

from imgutils.restore import remove_adversarial_noise

from plot import image_plot

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'),

        columns=2,

        figsize=(10, 6),

    )