module DeepViz

using PlotUtils
using Plots
using Zygote
using Zygote: hook, @adjoint
using Flux
using StatsBase
using Augmentor
using Images
using Metalhead

include("utils.jl")
include("vanilla_gradient.jl")
include("gradcam.jl")
include("guided_backprop.jl")
include("smooth_grad.jl")
include("image_generation.jl")

end
