esrgan: add model download for testing

Author: Acly

models/CMakeLists.txt

@@ -1,5 +1,4 @@
-# Download models used in tests
-# (this is disabled unless VISP_TESTS is enabled)
+# Download models used in tests (happens only if VISP_TESTS is enabled)
 
 message(STATUS "Checking for models/MobileSAM-F16.gguf")
 file(DOWNLOAD
@@ -22,4 +21,10 @@ file(DOWNLOAD
   EXPECTED_HASH "SHA256=c9f241e96fb5a791f9494fc7d4c2dd793297ae95f05b8423f547d19bea465b81"
   SHOW_PROGRESS
 )
-# TODO: ESRGAN
+message(STATUS "Checking for models/RealESRGAN-x4plus_anime-6B-F16.gguf")
+file(DOWNLOAD
+  "https://huggingface.co/Acly/Real-ESRGAN-GGUF/resolve/main/RealESRGAN-x4plus_anime-6B-F16.gguf"
+  ${CMAKE_CURRENT_LIST_DIR}/RealESRGAN-x4plus_anime-6B-F16.gguf
+  EXPECTED_HASH "SHA256=b741e68720d7ad6251dee2120bf7579ef816ea16da18299b39f6cbcb0e13ecf0"
+  SHOW_PROGRESS
+)

tests/test-models.cpp

@@ -1,7 +1,6 @@
 #include "util/string.hpp"
 #include "visp/vision.hpp"
 
-
 #include "testing.hpp"
 
 namespace visp {
@@ -48,7 +47,8 @@ VISP_BACKEND_TEST(test_birefnet)(backend_type bt) {
     image_data input = image_load(input_path.string().c_str());
     image_data output = birefnet_compute(model, input);
 
-    compare_images(name, output);
+    float tolerance = bt == backend_type::cpu ? 0.01f : 0.3f; // TODO: GPU is non-deterministic
+    compare_images(name, output, tolerance);
 }
 
 VISP_BACKEND_TEST(test_migan)(backend_type bt) {
@@ -69,7 +69,7 @@ VISP_BACKEND_TEST(test_migan)(backend_type bt) {
 }
 
 VISP_BACKEND_TEST(test_esrgan)(backend_type bt) {
-    path model_path = test_dir().models / "RealESRGAN_x4plus_anime_6Bh.gguf";
+    path model_path = test_dir().models / "RealESRGAN-x4plus_anime-6B-F16.gguf";
     path input_path = test_dir().input / "vase-and-bowl.jpg";
     std::string name = "esrgan";
     name += bt == backend_type::cpu ? "-cpu.png" : "-gpu.png";