{"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"name":"python","version":"3.10.12","mimetype":"text/x-python","codemirror_mode":{"name":"ipython","version":3},"pygments_lexer":"ipython3","nbconvert_exporter":"python","file_extension":".py"},"kaggle":{"accelerator":"gpu","dataSources":[{"sourceId":84969,"databundleVersionId":10033515,"sourceType":"competition"},{"sourceId":10665480,"sourceType":"datasetVersion","datasetId":6605277},{"sourceId":10665911,"sourceType":"datasetVersion","datasetId":6605496},{"sourceId":220858915,"sourceType":"kernelVersion"},{"sourceId":139474,"sourceType":"modelInstanceVersion","modelInstanceId":118113,"modelId":141350}],"dockerImageVersionId":30839,"isInternetEnabled":true,"language":"python","sourceType":"notebook","isGpuEnabled":true}},"nbformat_minor":4,"nbformat":4,"cells":[{"cell_type":"markdown","source":"## Install and Import Modules","metadata":{}},{"cell_type":"code","source":"!tar xfvz /kaggle/input/ultralytics-packages/archieve.tar.gz\n!pip install --no-index --find-links=./packages ultralytics\n!rm -rf ./packages","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-02-05T03:38:07.361459Z","iopub.execute_input":"2025-02-05T03:38:07.361788Z","iopub.status.idle":"2025-02-05T03:39:01.786559Z","shell.execute_reply.started":"2025-02-05T03:38:07.361764Z","shell.execute_reply":"2025-02-05T03:39:01.785637Z"},"_kg_hide-output":true},"outputs":[],"execution_count":null},{"cell_type":"code","source":"from tqdm import tqdm\nimport glob, os\nfrom ultralytics import YOLO","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-02-05T03:39:01.787626Z","iopub.execute_input":"2025-02-05T03:39:01.787944Z","iopub.status.idle":"2025-02-05T03:39:06.174113Z","shell.execute_reply.started":"2025-02-05T03:39:01.787911Z","shell.execute_reply":"2025-02-05T03:39:06.173178Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"#Load a pretrained model\nmodel = YOLO(\"/kaggle/input/yolo11/pytorch/default/1/yolo11l.pt\")","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-02-05T03:39:06.175035Z","iopub.execute_input":"2025-02-05T03:39:06.175442Z","iopub.status.idle":"2025-02-05T03:39:07.826049Z","shell.execute_reply.started":"2025-02-05T03:39:06.175420Z","shell.execute_reply":"2025-02-05T03:39:07.825110Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"_ = model.train(\n    data=\"/kaggle/input/czii-yolo-ddatasets/czii_conf.yaml\",\n    epochs=1000,\n    warmup_epochs=10,\n    optimizer='AdamW',\n    cos_lr=True,\n    lr0=3e-4,\n    lrf=0.03,\n    imgsz=640,\n    #device=\"0,1\",\n    weight_decay=0.005,\n    device=\"cuda\",\n    batch=16,\n    scale=0,\n    flipud=0.5,\n    fliplr=0.5,\n    degrees=45,\n    shear=5,\n    mixup=0.2,\n    copy_paste=0.25,\n    seed=5225,\n)","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-02-05T03:39:07.827018Z","iopub.execute_input":"2025-02-05T03:39:07.827356Z","iopub.status.idle":"2025-02-05T03:52:09.805503Z","shell.execute_reply.started":"2025-02-05T03:39:07.827331Z","shell.execute_reply":"2025-02-05T03:52:09.804619Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"model = YOLO(\"/kaggle/working/runs/detect/train/weights/best.pt\")\nmetrics = model.val(data=\"/kaggle/input/czii-yolo-ddatasets/czii_conf.yaml\", imgsz=640, batch=16, conf=0.25, iou=0.6, device=\"0\", save_json=True)  # no arguments needed, dataset and settings remembered\nprint(metrics.box.map)  # map50-95\nprint(metrics.box.map50)  # map50\nprint(metrics.box.map75)  # map75\nprint(metrics.box.maps)","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-02-05T03:53:42.789152Z","iopub.execute_input":"2025-02-05T03:53:42.789496Z","iopub.status.idle":"2025-02-05T03:53:53.898135Z","shell.execute_reply.started":"2025-02-05T03:53:42.789470Z","shell.execute_reply":"2025-02-05T03:53:53.897167Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"results = model(\"/kaggle/input/czii-yolo-ddatasets/datasets/czii_det2d/images/val/TS_5_4_920.png\")\nresults[0].show()","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-02-05T03:54:10.638142Z","iopub.execute_input":"2025-02-05T03:54:10.638495Z","iopub.status.idle":"2025-02-05T03:54:10.812381Z","shell.execute_reply.started":"2025-02-05T03:54:10.638464Z","shell.execute_reply":"2025-02-05T03:54:10.811567Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"results = model(\"/kaggle/input/czii-yolo-ddatasets/datasets/czii_det2d/images/val/TS_5_4_1240.png\")\nresults[0].show()","metadata":{"trusted":true},"outputs":[],"execution_count":null}]}