{"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":"markdown","source":"Dataset Link - https://www.kaggle.com/datasets/sashanktalakola/ubc-ocean-resize-dataset","metadata":{}},{"cell_type":"code","source":"import os\nimport pandas as pd\nimport cv2\n\ndf = pd.read_csv(\"/kaggle/input/UBC-OCEAN/train.csv\")\ndf.head()","metadata":{"_uuid":"8f2839f25d086af736a60e9eeb907d3b93b6e0e5","_cell_guid":"b1076dfc-b9ad-4769-8c92-a6c4dae69d19","trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"IMG_SIZES = [256, 512, 1024]\n\nfor img_size in IMG_SIZES:\n    os.mkdir(f\"{img_size}_{img_size}\")","metadata":{"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"def resize_img(image, img_size):\n    resized_image = cv2.resize(image, (img_size, img_size))\n    return resized_image","metadata":{"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"for row in df.iterrows():\n    image_id, is_tma = row[1].image_id, row[1].is_tma\n    if is_tma:\n        img_path = os.path.join(\"/kaggle/input/UBC-OCEAN/train_images\", f\"{image_id}.png\")\n    else:\n        img_path = os.path.join(\"/kaggle/input/UBC-OCEAN/train_thumbnails\", f\"{image_id}_thumbnail.png\")\n        \n    image = cv2.imread(img_path)\n    for img_size in IMG_SIZES:\n        resized_image = resize_img(image, img_size)\n        cv2.imwrite(os.path.join(f\"{img_size}_{img_size}\", f'{image_id}.png'), resized_image)","metadata":{"trusted":true},"execution_count":null,"outputs":[]}]}