Remove ImageNet preprocessing utilities (#159)

adrhill · web-flow · commit 0fb4bd3f5e65 · 2024-01-26T21:57:34.000+01:00
diff --git a/README.md b/README.md
@@ -40,10 +40,7 @@ model = strip_softmax(model)
 model = canonize(model)
 
 # Load input
-url = HTTP.URI("https://raw.githubusercontent.com/Julia-XAI/ExplainableAI.jl/gh-pages/assets/heatmaps/castle.jpg")
-img = load(url)
-input = preprocess_imagenet(img)
-input = reshape(input, 224, 224, 3, :)  # reshape to WHCN format
+input = ...                       # input in WHCN format
 
 # Run XAI method
 composite = EpsilonPlusFlat()
diff --git a/docs/src/api.md b/docs/src/api.md
@@ -23,11 +23,6 @@ NoiseAugmentation
 InterpolationAugmentation
 ```
 
-# Input preprocessing
-```@docs
-preprocess_imagenet
-```
-
 # Index
 ```@index
 ```
diff --git a/src/ExplainableAI.jl b/src/ExplainableAI.jl
@@ -15,11 +15,9 @@ include("bibliography.jl")
 include("utils.jl")
 include("input_augmentation.jl")
 include("gradient.jl")
-include("preprocessing.jl")
 
 export Gradient, InputTimesGradient
 export NoiseAugmentation, SmoothGrad
 export InterpolationAugmentation, IntegratedGradients
-export preprocess_imagenet
 
 end # module
diff --git a/src/preprocessing.jl b/src/preprocessing.jl
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -21,10 +21,6 @@ pseudorand(dims...) = rand(MersenneTwister(123), Float32, dims...)
         @info "Testing input augmentation..."
         include("test_input_augmentation.jl")
     end
-    @testset "ImageNet preprocessing" begin
-        @info "Testing ImageNet preprocessing..."
-        include("test_imagenet.jl")
-    end
     @testset "CNN" begin
         @info "Testing analyzers on CNN..."
         include("test_cnn.jl")
diff --git a/test/test_imagenet.jl b/test/test_imagenet.jl
