Showing
1 changed file
with
82 additions
and
0 deletions
notebooks/resize.ipynb
0 → 100644
| 1 | +{ | ||
| 2 | + "cells": [ | ||
| 3 | + { | ||
| 4 | + "cell_type": "code", | ||
| 5 | + "execution_count": 1, | ||
| 6 | + "id": "ahead-paste", | ||
| 7 | + "metadata": {}, | ||
| 8 | + "outputs": [], | ||
| 9 | + "source": [ | ||
| 10 | + "from glob import glob\n", | ||
| 11 | + "import cv2\n", | ||
| 12 | + "\n", | ||
| 13 | + "images = sorted(glob(\"../bbb_sunflower_540p/*.png\"))" | ||
| 14 | + ] | ||
| 15 | + }, | ||
| 16 | + { | ||
| 17 | + "cell_type": "code", | ||
| 18 | + "execution_count": 2, | ||
| 19 | + "id": "rapid-tension", | ||
| 20 | + "metadata": {}, | ||
| 21 | + "outputs": [], | ||
| 22 | + "source": [ | ||
| 23 | + "from pathlib import Path\n", | ||
| 24 | + "Path(\"../dataset/DIV2K/DIV2K_train_HR\").mkdir(parents=True, exist_ok=True)\n", | ||
| 25 | + "Path(\"../dataset/DIV2K/DIV2K_train_LR_bicubic/X2\").mkdir(parents=True, exist_ok=True)" | ||
| 26 | + ] | ||
| 27 | + }, | ||
| 28 | + { | ||
| 29 | + "cell_type": "code", | ||
| 30 | + "execution_count": 3, | ||
| 31 | + "id": "visible-texas", | ||
| 32 | + "metadata": {}, | ||
| 33 | + "outputs": [ | ||
| 34 | + { | ||
| 35 | + "name": "stderr", | ||
| 36 | + "output_type": "stream", | ||
| 37 | + "text": [ | ||
| 38 | + "100%|██████████| 1800/1800 [01:21<00:00, 22.00it/s]\n" | ||
| 39 | + ] | ||
| 40 | + } | ||
| 41 | + ], | ||
| 42 | + "source": [ | ||
| 43 | + "from tqdm import tqdm\n", | ||
| 44 | + "for image in tqdm(images):\n", | ||
| 45 | + " hr = cv2.imread(image, cv2.IMREAD_COLOR)\n", | ||
| 46 | + " lr = cv2.resize(hr, dsize=(480, 270), interpolation=cv2.INTER_CUBIC)\n", | ||
| 47 | + "\n", | ||
| 48 | + " cv2.imwrite(\"../dataset/DIV2K/DIV2K_train_HR/\" + Path(image).name, hr)\n", | ||
| 49 | + " cv2.imwrite(\"../dataset/DIV2K/DIV2K_train_LR_bicubic/X2/\" + Path(image).stem + \"x2.png\", lr)" | ||
| 50 | + ] | ||
| 51 | + }, | ||
| 52 | + { | ||
| 53 | + "cell_type": "code", | ||
| 54 | + "execution_count": null, | ||
| 55 | + "id": "fallen-religion", | ||
| 56 | + "metadata": {}, | ||
| 57 | + "outputs": [], | ||
| 58 | + "source": [] | ||
| 59 | + } | ||
| 60 | + ], | ||
| 61 | + "metadata": { | ||
| 62 | + "kernelspec": { | ||
| 63 | + "display_name": "Python 3", | ||
| 64 | + "language": "python", | ||
| 65 | + "name": "python3" | ||
| 66 | + }, | ||
| 67 | + "language_info": { | ||
| 68 | + "codemirror_mode": { | ||
| 69 | + "name": "ipython", | ||
| 70 | + "version": 3 | ||
| 71 | + }, | ||
| 72 | + "file_extension": ".py", | ||
| 73 | + "mimetype": "text/x-python", | ||
| 74 | + "name": "python", | ||
| 75 | + "nbconvert_exporter": "python", | ||
| 76 | + "pygments_lexer": "ipython3", | ||
| 77 | + "version": "3.7.7" | ||
| 78 | + } | ||
| 79 | + }, | ||
| 80 | + "nbformat": 4, | ||
| 81 | + "nbformat_minor": 5 | ||
| 82 | +} |
-
Please register or login to post a comment