{"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"name":"python","version":"3.11.13","mimetype":"text/x-python","codemirror_mode":{"name":"ipython","version":3},"pygments_lexer":"ipython3","nbconvert_exporter":"python","file_extension":".py"},"kaggle":{"accelerator":"nvidiaTeslaT4","dataSources":[{"sourceId":107469,"databundleVersionId":13058354,"isSourceIdPinned":false,"sourceType":"competition"},{"sourceId":12896512,"sourceType":"datasetVersion","datasetId":8159780}],"dockerImageVersionId":31090,"isInternetEnabled":true,"language":"python","sourceType":"notebook","isGpuEnabled":true}},"nbformat_minor":4,"nbformat":4,"cells":[{"cell_type":"code","source":"from IPython.display import clear_output\n!pip install ultralytics\nclear_output()","metadata":{"_uuid":"8f2839f25d086af736a60e9eeb907d3b93b6e0e5","_cell_guid":"b1076dfc-b9ad-4769-8c92-a6c4dae69d19","trusted":true,"execution":{"iopub.status.busy":"2025-08-28T13:11:35.069219Z","iopub.execute_input":"2025-08-28T13:11:35.069379Z","iopub.status.idle":"2025-08-28T13:12:56.795419Z","shell.execute_reply.started":"2025-08-28T13:11:35.069363Z","shell.execute_reply":"2025-08-28T13:12:56.794531Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"from ultralytics import YOLO\nfrom pathlib import Path\nimport csv\nimport os\nimport numpy as np\nimport pandas as pd\nimport random\nimport torch\n# Set random seeds for reproducibility\nnp.random.seed(42)\nrandom.seed(42)\ntorch.manual_seed(42)","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-08-28T13:12:56.796722Z","iopub.execute_input":"2025-08-28T13:12:56.797046Z","iopub.status.idle":"2025-08-28T13:13:01.814726Z","shell.execute_reply.started":"2025-08-28T13:12:56.797022Z","shell.execute_reply":"2025-08-28T13:13:01.814066Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"data_yaml='''\npath: /kaggle/input/falcon-multiclass-dataset\n\ntrain:  [dataset/fold_2/images, dataset/fold_3/images, dataset/fold_4/images, dataset/fold_5/images]\nval:  [dataset/fold_1/images, dataset/val/images]\ntest: /kaggle/input/multi-class-object-detection-challenge/testImages/images\nnc: 2\nnames: ['cheerios', 'soup']\n'''\nwith open('data1.yaml', 'w') as file:\n    file.write(data_yaml)","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-08-28T13:13:01.819158Z","iopub.execute_input":"2025-08-28T13:13:01.819852Z","iopub.status.idle":"2025-08-28T13:13:01.823687Z","shell.execute_reply.started":"2025-08-28T13:13:01.819825Z","shell.execute_reply":"2025-08-28T13:13:01.823066Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"data_yaml='''\npath: /kaggle/input/falcon-multiclass-dataset\n\ntrain:  [dataset/fold_1/images, dataset/fold_3/images, dataset/fold_4/images, dataset/fold_5/images]\nval:  [dataset/fold_2/images, dataset/val/images]\ntest: /kaggle/input/multi-class-object-detection-challenge/testImages/images\nnc: 2\nnames: ['cheerios', 'soup']\n'''\nwith open('data2.yaml', 'w') as file:\n    file.write(data_yaml)","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-08-28T13:13:01.824631Z","iopub.execute_input":"2025-08-28T13:13:01.824892Z","iopub.status.idle":"2025-08-28T13:13:04.416974Z","shell.execute_reply.started":"2025-08-28T13:13:01.824869Z","shell.execute_reply":"2025-08-28T13:13:04.416202Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"data_yaml='''\npath: /kaggle/input/falcon-multiclass-dataset\n\ntrain:  [dataset/fold_1/images, dataset/fold_2/images, dataset/fold_4/images, dataset/fold_5/images]\nval:  [dataset/fold_3/images, dataset/val/images]\ntest: /kaggle/input/multi-class-object-detection-challenge/testImages/images\nnc: 2\nnames: ['cheerios', 'soup']\n'''\nwith open('data3.yaml', 'w') as file:\n    file.write(data_yaml)","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-08-28T13:13:04.417810Z","iopub.execute_input":"2025-08-28T13:13:04.418205Z","iopub.status.idle":"2025-08-28T13:13:04.431324Z","shell.execute_reply.started":"2025-08-28T13:13:04.418178Z","shell.execute_reply":"2025-08-28T13:13:04.430833Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"data_yaml='''\npath: /kaggle/input/falcon-multiclass-dataset\n\ntrain:  [dataset/fold_1/images, dataset/fold_2/images, dataset/fold_3/images, dataset/fold_5/images]\nval:  [dataset/fold_4/images, dataset/val/images]\ntest: /kaggle/input/multi-class-object-detection-challenge/testImages/images\nnc: 2\nnames: ['cheerios', 'soup']\n'''\nwith open('data4.yaml', 'w') as file:\n    file.write(data_yaml)","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-08-28T13:13:04.432023Z","iopub.execute_input":"2025-08-28T13:13:04.432587Z","iopub.status.idle":"2025-08-28T13:13:04.446010Z","shell.execute_reply.started":"2025-08-28T13:13:04.432569Z","shell.execute_reply":"2025-08-28T13:13:04.445508Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"data_yaml='''\npath: /kaggle/input/falcon-multiclass-dataset\n\ntrain:  [dataset/fold_1/images, dataset/fold_2/images, dataset/fold_3/images, dataset/fold_4/images]\nval:  [dataset/fold_5/images, dataset/val/images]\ntest: /kaggle/input/multi-class-object-detection-challenge/testImages/images\nnc: 2\nnames: ['cheerios', 'soup']\n'''\nwith open('data5.yaml', 'w') as file:\n    file.write(data_yaml)","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-08-28T13:13:04.446659Z","iopub.execute_input":"2025-08-28T13:13:04.446887Z","iopub.status.idle":"2025-08-28T13:13:04.463088Z","shell.execute_reply.started":"2025-08-28T13:13:04.446872Z","shell.execute_reply":"2025-08-28T13:13:04.462341Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"model = YOLO(\"yolo11x.pt\")\n\nfor i in range(1,6):\n    \n    model.train(\n        data=f\"data{i}.yaml\", \n        epochs=40,                \n        batch=8,                   \n        imgsz=1056,\n        patience=40,               \n        optimizer='SGD',\n        momentum=0.937,          \n        lr0=0.0002,                \n        weight_decay=0.0005,       \n        cos_lr=True,               \n        save_period=5,             \n        workers=8,\n        device=[0, 1],\n        # Augmentations\n        close_mosaic=0,\n        hsv_h=0.015,\n        hsv_s=0.7,\n        hsv_v=0.4,\n        flipud=0,\n        fliplr=0.5,\n        translate=0.1,\n        scale=0.5,\n        shear=0.01,\n        mixup=0.1,\n        warmup_epochs= 3,\n        # warmup_momentum= 1,\n        cache=True,\n        amp=True,\n    )\n","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-08-28T13:13:04.463921Z","iopub.execute_input":"2025-08-28T13:13:04.464223Z","execution_failed":"2025-08-28T13:19:21.654Z"}},"outputs":[],"execution_count":null}]}