{"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"},"kaggle":{"accelerator":"nvidiaTeslaT4","dataSources":[{"sourceId":117876,"databundleVersionId":14198377,"sourceType":"competition"},{"sourceId":271554247,"sourceType":"kernelVersion"}],"dockerImageVersionId":31193,"isInternetEnabled":true,"language":"python","sourceType":"notebook","isGpuEnabled":true}},"nbformat_minor":4,"nbformat":4,"cells":[{"cell_type":"code","source":"!pip install ultralytics > /dev/null\n!pip install \"numpy<2\" --upgrade --force-reinstall > /dev/null","metadata":{"_uuid":"8f2839f25d086af736a60e9eeb907d3b93b6e0e5","_cell_guid":"b1076dfc-b9ad-4769-8c92-a6c4dae69d19","trusted":true,"_kg_hide-output":true},"outputs":[],"execution_count":null},{"cell_type":"code","source":"import pandas as pd\nfrom pathlib import Path\nfrom ultralytics import YOLO\n\ndata_yaml = \"\"\"\ntrain: /kaggle/input/the-3lc-cotton-weed-detection-challenge/cotton_weed_competition_dataset/train/images\nval: /kaggle/input/the-3lc-cotton-weed-detection-challenge/cotton_weed_competition_dataset/val\n\nnc: 3\nnames: ['Carpetweed', 'MorningGlory', 'PalmerAmaranth']\n\"\"\"\n\nwith open('data.yaml', 'w') as file:\n    file.write(data_yaml)","metadata":{"trusted":true},"outputs":[],"execution_count":null},{"cell_type":"code","source":"model = YOLO(\"/kaggle/input/weed-detection-model/runs/train/train/weights/best.pt\")\n\nmodel.train(\n    data=\"data.yaml\",\n    epochs=30,\n    batch=32,\n    imgsz=640,\n    device=[0, 1],\n    \n    patience=10,\n    optimizer='SGD',\n    lr0=0.00020031238264196197,\n    lrf=0.1008095673904864,\n    momentum=0.8550324212523863,\n    weight_decay=0.000207090196079648,\n    \n    augment=True,\n    mosaic=0.6634492032540517,\n    mixup=0.04061951669770144,\n    copy_paste=0.08419861026499094,\n    hsv_h=0.0655157018230224,\n    hsv_s=0.049289892615726576,\n    hsv_v=0.2253605011673964,\n    degrees=5.8617569056467715,\n    translate=0.3520982284315102,\n    scale=0.7873805608762232,\n    shear=2.5754850262711666,\n    perspective=0.0002011629103742282,\n    fliplr=0.5342482519791614,\n    flipud=0.12264350361235729,\n    \n    warmup_epochs=2,\n    warmup_momentum=0.8,\n    warmup_bias_lr=0.1,\n    \n    close_mosaic=10,\n    dropout=0.029532543969579855,\n    seed=52,\n    pretrained=False,\n    overlap_mask=True,\n    nbs=64,\n    save_period=2,\n    amp=True,\n    workers=8,\n)","metadata":{"trusted":true},"outputs":[],"execution_count":null}]}