{"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"name":"python","version":"3.10.14","mimetype":"text/x-python","codemirror_mode":{"name":"ipython","version":3},"pygments_lexer":"ipython3","nbconvert_exporter":"python","file_extension":".py"},"kaggle":{"accelerator":"none","dataSources":[{"sourceId":13836,"databundleVersionId":1718836,"sourceType":"competition"},{"sourceId":9969858,"sourceType":"datasetVersion","datasetId":6133487},{"sourceId":9970102,"sourceType":"datasetVersion","datasetId":6133676},{"sourceId":173786,"sourceType":"modelInstanceVersion","isSourceIdPinned":true,"modelInstanceId":147930,"modelId":170455},{"sourceId":187750,"sourceType":"modelInstanceVersion","isSourceIdPinned":true,"modelInstanceId":160065,"modelId":182441},{"sourceId":189774,"sourceType":"modelInstanceVersion","isSourceIdPinned":true,"modelInstanceId":161801,"modelId":184169}],"dockerImageVersionId":30786,"isInternetEnabled":false,"language":"python","sourceType":"notebook","isGpuEnabled":false}},"nbformat_minor":4,"nbformat":4,"cells":[{"cell_type":"code","source":"!pip install ultralytics --no-index --find-links=file:///kaggle/input/ultralytics/ultralytics","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2024-12-05T23:41:24.211034Z","iopub.execute_input":"2024-12-05T23:41:24.211742Z","iopub.status.idle":"2024-12-05T23:41:37.759811Z","shell.execute_reply.started":"2024-12-05T23:41:24.211682Z","shell.execute_reply":"2024-12-05T23:41:37.758406Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"from csv import writer\nfrom ultralytics import YOLO\nimport os","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2024-12-05T23:41:37.762572Z","iopub.execute_input":"2024-12-05T23:41:37.762967Z","iopub.status.idle":"2024-12-05T23:41:44.079725Z","shell.execute_reply.started":"2024-12-05T23:41:37.762930Z","shell.execute_reply":"2024-12-05T23:41:44.078581Z"}},"outputs":[],"execution_count":null},{"cell_type":"markdown","source":"## Predict","metadata":{}},{"cell_type":"code","source":"!rm /kaggle/working/submission.csv","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2024-12-05T23:41:44.081032Z","iopub.execute_input":"2024-12-05T23:41:44.081541Z","iopub.status.idle":"2024-12-05T23:41:45.264056Z","shell.execute_reply.started":"2024-12-05T23:41:44.081505Z","shell.execute_reply":"2024-12-05T23:41:45.262317Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"predict_model = YOLO(\"//kaggle/input/pretrained-large-epoch/other/default/1/best.pt\")\n\nimg_list = []\n\n\nwith open(\"submission.csv\", \"a\") as f:\n    writer_obj = writer(f)\n    writer_obj.writerow([\"image_id\", \"label\"])\n    f.close()\n\nfor image in list(os.listdir(\"/kaggle/input/cassava-leaf-disease-classification/test_images\")):\n    img_list.append('/kaggle/input/cassava-leaf-disease-classification/test_images/' + image)\n\nwith open(\"submission.csv\", \"a\") as f:\n    writer_obj = writer(f)\n    for image in img_list:  \n        result = predict_model(image)\n        name = predict_model.names\n        if str(result[0].boxes.cls) == 'tensor([])':\n            label = 3\n        else:\n            label = int(result[0].boxes.cls[0])\n        writer_obj.writerow([result[0].path[62:], name[label]])\n    f.close()\n","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2024-12-05T23:41:45.268140Z","iopub.execute_input":"2024-12-05T23:41:45.268797Z","iopub.status.idle":"2024-12-05T23:41:48.444621Z","shell.execute_reply.started":"2024-12-05T23:41:45.268754Z","shell.execute_reply":"2024-12-05T23:41:48.443369Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"!head submission.csv","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2024-12-05T23:41:48.446128Z","iopub.execute_input":"2024-12-05T23:41:48.446676Z","iopub.status.idle":"2024-12-05T23:41:49.632115Z","shell.execute_reply.started":"2024-12-05T23:41:48.446640Z","shell.execute_reply":"2024-12-05T23:41:49.630442Z"}},"outputs":[],"execution_count":null}]}