Skip to content

Implementation of image optimization based on any neural network's outputs

Notifications You must be signed in to change notification settings

Farbod-Siahkali/Input-Image-Optimizer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

15 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Input Optimizer

Input Optimizer is a PyTorch code designed to optimize input images for any deep learning network based on a specific target. The code was developed in the Human and Robot Interaction Laboratory (Taarlab).

Optimizing Input Noise on Mark1501 Dataset

One of the library's features is the ability to optimize input noise going through a pre-trained network on Mark1501 dataset. The following images demonstrate how the library optimizes based on a specific attribute:

c_5_iter_999_loss_-26 289833068847656 | c_30_iter_758_loss_-29 52212142944336 | c_25_iter_448_loss_-22 735559463500977

Optimizing Input Noise on Pre-trained AlexNet Model

Another feature of Input Optimizer is the ability to optimize input noise going through a pre-trained AlexNet model. The following images demonstrate how the library optimizes based on a specific attribute:

Requirements

To use Input Optimizer, you need to have the following Python packages installed:

pythonCopy code
torch torchvision copy numpy PIL

For more information, please contact the developer at [email protected] or [email protected].

About

Implementation of image optimization based on any neural network's outputs

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages