{"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"name":"python","version":"3.10.13","mimetype":"text/x-python","codemirror_mode":{"name":"ipython","version":3},"pygments_lexer":"ipython3","nbconvert_exporter":"python","file_extension":".py"},"kaggle":{"accelerator":"gpu","dataSources":[{"sourceId":13836,"databundleVersionId":1718836,"sourceType":"competition"}],"dockerImageVersionId":30699,"isInternetEnabled":true,"language":"python","sourceType":"notebook","isGpuEnabled":true}},"nbformat_minor":4,"nbformat":4,"cells":[{"cell_type":"code","source":"import pandas as pd\nimport cv2\nimport os \nimport torch.nn as nn\nfrom torch.utils.data import Dataset ,DataLoader, Subset\nimport torch.optim as optim\nimport numpy as np\nimport torch\nfrom torchvision import transforms\nimport matplotlib.pyplot as plt\nfrom PIL import Image\nfrom torch.utils.data import random_split\nfrom tqdm import tqdm","metadata":{"_uuid":"8f2839f25d086af736a60e9eeb907d3b93b6e0e5","_cell_guid":"b1076dfc-b9ad-4769-8c92-a6c4dae69d19","execution":{"iopub.status.busy":"2024-05-19T00:56:08.181006Z","iopub.execute_input":"2024-05-19T00:56:08.181702Z","iopub.status.idle":"2024-05-19T00:56:08.187151Z","shell.execute_reply.started":"2024-05-19T00:56:08.18167Z","shell.execute_reply":"2024-05-19T00:56:08.186083Z"},"trusted":true},"execution_count":3,"outputs":[]},{"cell_type":"code","source":"data_dir = '/kaggle/input/cassava-leaf-disease-classification'\nbatch_size = 32\nepochs = 30\nnum_classes = 5\nlr_rate = 0.05\n\ndevice = torch.device(\"cuda:0\" if torch.cuda.is_available() else \"cpu\")","metadata":{"execution":{"iopub.status.busy":"2024-05-19T00:56:13.940803Z","iopub.execute_input":"2024-05-19T00:56:13.941438Z","iopub.status.idle":"2024-05-19T00:56:13.946231Z","shell.execute_reply.started":"2024-05-19T00:56:13.941404Z","shell.execute_reply":"2024-05-19T00:56:13.945286Z"},"trusted":true},"execution_count":5,"outputs":[]},{"cell_type":"code","source":"os.listdir(data_dir)\nprint('Train images: %d' %len(os.listdir(os.path.join(data_dir, \"train_images\"))))","metadata":{"execution":{"iopub.status.busy":"2024-05-19T00:56:19.675309Z","iopub.execute_input":"2024-05-19T00:56:19.675749Z","iopub.status.idle":"2024-05-19T00:56:19.959656Z","shell.execute_reply.started":"2024-05-19T00:56:19.675715Z","shell.execute_reply":"2024-05-19T00:56:19.958629Z"},"trusted":true},"execution_count":6,"outputs":[{"name":"stdout","text":"Train images: 21397\n","output_type":"stream"}]},{"cell_type":"code","source":"train_labels = pd.read_csv(os.path.join(data_dir, \"train.csv\"))\ntrain_labels.head()","metadata":{"execution":{"iopub.status.busy":"2024-05-19T00:56:25.370928Z","iopub.execute_input":"2024-05-19T00:56:25.371545Z","iopub.status.idle":"2024-05-19T00:56:25.414171Z","shell.execute_reply.started":"2024-05-19T00:56:25.371515Z","shell.execute_reply":"2024-05-19T00:56:25.413155Z"},"trusted":true},"execution_count":7,"outputs":[{"execution_count":7,"output_type":"execute_result","data":{"text/plain":" image_id label\n0 1000015157.jpg 0\n1 1000201771.jpg 3\n2 100042118.jpg 1\n3 1000723321.jpg 1\n4 1000812911.jpg 3","text/html":"\n\n
\n \n \n | \n image_id | \n label | \n
\n \n \n \n | 0 | \n 1000015157.jpg | \n 0 | \n
\n \n | 1 | \n 1000201771.jpg | \n 3 | \n
\n \n | 2 | \n 100042118.jpg | \n 1 | \n
\n \n | 3 | \n 1000723321.jpg | \n 1 | \n
\n \n | 4 | \n 1000812911.jpg | \n 3 | \n
\n \n
\n