Showing
1 changed file
with
72 additions
and
0 deletions
notebooks/psnr.ipynb
0 → 100644
1 | +{ | ||
2 | + "cells": [ | ||
3 | + { | ||
4 | + "cell_type": "code", | ||
5 | + "execution_count": 15, | ||
6 | + "id": "automotive-circus", | ||
7 | + "metadata": {}, | ||
8 | + "outputs": [], | ||
9 | + "source": [ | ||
10 | + "from glob import glob\n", | ||
11 | + "import cv2\n", | ||
12 | + "from tqdm import tqdm\n", | ||
13 | + "\n", | ||
14 | + "gt_list = sorted(glob(\"../bbb_sunflower_1080p/*.png\"))\n", | ||
15 | + "sr_list = sorted(glob(\"./experiment/test/results-Demo/*.png\"))" | ||
16 | + ] | ||
17 | + }, | ||
18 | + { | ||
19 | + "cell_type": "code", | ||
20 | + "execution_count": 16, | ||
21 | + "id": "unique-wichita", | ||
22 | + "metadata": {}, | ||
23 | + "outputs": [ | ||
24 | + { | ||
25 | + "name": "stderr", | ||
26 | + "output_type": "stream", | ||
27 | + "text": [ | ||
28 | + "100%|██████████| 1800/1800 [02:55<00:00, 10.24it/s]\n" | ||
29 | + ] | ||
30 | + } | ||
31 | + ], | ||
32 | + "source": [ | ||
33 | + "with open(\"result.csv\", \"w\") as f:\n", | ||
34 | + " f.write(\"idx,psnr\\n\");\n", | ||
35 | + " for idx, (gt, sr) in tqdm(enumerate(zip(gt_list, sr_list)), total=len(sr_list)):\n", | ||
36 | + " gt_img = cv2.imread(gt)\n", | ||
37 | + " sr_img = cv2.imread(sr)\n", | ||
38 | + " psnr = cv2.PSNR(gt_img, sr_img)\n", | ||
39 | + " f.write(f\"{idx},{psnr}\\n\")" | ||
40 | + ] | ||
41 | + }, | ||
42 | + { | ||
43 | + "cell_type": "code", | ||
44 | + "execution_count": null, | ||
45 | + "id": "raised-projector", | ||
46 | + "metadata": {}, | ||
47 | + "outputs": [], | ||
48 | + "source": [] | ||
49 | + } | ||
50 | + ], | ||
51 | + "metadata": { | ||
52 | + "kernelspec": { | ||
53 | + "display_name": "Python 3", | ||
54 | + "language": "python", | ||
55 | + "name": "python3" | ||
56 | + }, | ||
57 | + "language_info": { | ||
58 | + "codemirror_mode": { | ||
59 | + "name": "ipython", | ||
60 | + "version": 3 | ||
61 | + }, | ||
62 | + "file_extension": ".py", | ||
63 | + "mimetype": "text/x-python", | ||
64 | + "name": "python", | ||
65 | + "nbconvert_exporter": "python", | ||
66 | + "pygments_lexer": "ipython3", | ||
67 | + "version": "3.7.7" | ||
68 | + } | ||
69 | + }, | ||
70 | + "nbformat": 4, | ||
71 | + "nbformat_minor": 5 | ||
72 | +} |
-
Please register or login to post a comment