From 5c2576245de6cc8a80e09d5af022bb21fb2e2f8e Mon Sep 17 00:00:00 2001
From: Guillermo <guillermo.sahonero@gmail.com>
Date: Mon, 18 Nov 2024 17:48:01 -0300
Subject: [PATCH] Demo file in test folder

---
 test/demo.jl | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)
 create mode 100644 test/demo.jl

diff --git a/test/demo.jl b/test/demo.jl
new file mode 100644
index 0000000..14a89a3
--- /dev/null
+++ b/test/demo.jl
@@ -0,0 +1,19 @@
+using DISTS
+using Images
+
+# Read images
+ref = Images.load("../DISTS/images/r0.png")
+dist = Images.load("../DISTS/images/r1.png")
+
+# Load the pretrained network parameters and perceptual weights
+net_params, weights = load_weights("../DISTS/weights/net_param.mat", "../DISTS/weights/alpha_beta.mat")
+
+# Define resize image flag and use_gpu flag
+resize_img = true   # If required. Its usage is not trivial.
+use_gpu = false     # GPU acceleration is not implemented
+
+# Calculate the perceptual quality score (DISTS)
+@time score = DISTS_score(ref, dist, net_params, weights, resize_img, use_gpu; pretraining = "DISTS")
+
+# Output the score
+println("Perceptual quality score: ", score)
\ No newline at end of file