using MIPVerify
using Gurobi
using Images

mnist = MIPVerify.read_datasets("MNIST")

mnist.train

size(mnist.train.images)

colorview(Gray, mnist.train.images[1, :, :, 1])

mnist.train.labels

n1params = MIPVerify.get_example_network_params("MNIST.n1")

MIPVerify.frac_correct(n1params, mnist.test, 10000)

sample_image = MIPVerify.get_image(mnist.train.images, 1)

output_activations = sample_image |> n1params

(output_activations |> MIPVerify.get_max_index) - 1

MIPVerify.get_label(mnist.train.labels, 1)

target_label_index = 9
d = MIPVerify.find_adversarial_example(n1params, sample_image, target_label_index, GurobiSolver())

using JuMP

perturbed_sample_image = getvalue(d[:PerturbedInput])

perturbed_sample_image |> n1params

colorview(Gray, perturbed_sample_image[1, :, :, 1])

colorview(Gray, sample_image[1, :, :, 1])