{"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"pygments_lexer":"ipython3","nbconvert_exporter":"python","version":"3.6.4","file_extension":".py","codemirror_mode":{"name":"ipython","version":3},"name":"python","mimetype":"text/x-python"}},"nbformat_minor":4,"nbformat":4,"cells":[{"cell_type":"code","source":"import numpy as np\nimport pandas as pd\n\nimport torch\nimport torch.nn as nn\nimport torch.nn.functional as F\nfrom torch.utils.data import Dataset, DataLoader# For custom data-sets\n\nimport torchvision\nimport torchvision.transforms as T\n\nimport cv2 as cv\nimport numpy as np\nimport matplotlib.pyplot as plt\nimport os\nfrom PIL import Image\nimport seaborn as sns","metadata":{"_uuid":"8f2839f25d086af736a60e9eeb907d3b93b6e0e5","_cell_guid":"b1076dfc-b9ad-4769-8c92-a6c4dae69d19","trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"root_dir = '/kaggle/input/plant-pathology-2021-fgvc8/train_images'\nout_dir = 'resize-plant-{}'","metadata":{"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"def center_crop(img):\n    \"\"\"\n    1:1的中心切图\n    \"\"\"\n    w,h,c = img.shape\n    if w > h:\n        w0 = int((w-h)/2)\n        w1 = w0 + h\n        img = img[w0:w1,:]\n    else:\n        h0 = int((h-w)/2)\n        h1 = h0 + w\n        img = img[:,h0:h1]\n    return img","metadata":{"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"dsizes = [512, 256, 128, 64]\n\n# 输出目录检测\nfor dsize in dsizes:\n    resize_out_dir = out_dir.format(dsize)\n    if not os.path.exists(resize_out_dir):\n        os.mkdir(resize_out_dir)\n\n# 图片resize\nfiles = os.listdir(root_dir)\nfor file in files:\n    img_path = os.path.join(root_dir, file)\n    img = cv.imread(img_path)\n    img = center_crop(img)\n    \n    # 输出图片\n    for dsize in dsizes:\n        resize_out_dir = out_dir.format(dsize)\n        img_out_path = os.path.join(resize_out_dir, file)\n        img_resized = cv.resize(img, dsize=(dsize, dsize))\n        cv.imwrite(img_out_path, img_resized)","metadata":{"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"! cp /kaggle/input/plant-pathology-2021-fgvc8/train_images/train.csv .","metadata":{"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"! ls #resize-plant-512/","metadata":{"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"! du -sh *","metadata":{"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"# ! rm -rf resize-p*","metadata":{"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"","metadata":{},"execution_count":null,"outputs":[]}]}