Use test data for benchmarking

jakewilliami · Nov 5, 2024 · 0ce0891 · 0ce0891
1 parent 890f29f
commit 0ce0891
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 10 deletions.
diff --git a/benchmark/basic.jl b/benchmark/basic.jl
@@ -9,16 +9,7 @@ using BenchmarkTools: @btime
 
 Logging.disable_logging(Logging.Info)
 
-include(joinpath(dirname(@__DIR__), "examples", "constants.jl"))
-include(joinpath(dirname(@__DIR__), "examples", "main_data.jl"))
-
-min_size_img = (19, 19)
-scale, scale_to = false, (200, 200)
-
-min_feature_height = 8
-max_feature_height = 10
-min_feature_width = 8
-max_feature_width = 10
+include("constants.jl")
 
 learn(
     pos_training_path,

diff --git a/benchmark/constants.jl b/benchmark/constants.jl
@@ -0,0 +1,16 @@
+# Use same data as tests, for simplicity
+main_data_path = joinpath(dirname(@__DIR__), "test", "images")
+
+pos_training_path = joinpath(main_data_path, "pos")
+neg_training_path = joinpath(main_data_path, "neg")
+pos_testing_path = joinpath(main_data_path, "pos_testing")
+neg_testing_path = joinpath(main_data_path, "neg_testing")
+
+num_classifiers = 10
+min_size_img = (19, 19)
+scale, scale_to = false, (200, 200)
+
+min_feature_height = 8
+max_feature_height = 10
+min_feature_width = 8
+max_feature_width = 10