yolov8_object_detection_on_custom_dataset.ipynb

{"cells":[{"cell_type":"code","execution_count":null,"metadata":{"colab":{"base_uri":"https://localhost:8080/"},"executionInfo":{"elapsed":2,"status":"ok","timestamp":1680486210312,"user":{"displayName":"Freedom Tech","userId":"03870511654783100627"},"user_tz":-330},"id":"CjpPg4mGKc1v","outputId":"1c05430b-239f-47f7-f245-625a4b2a1633"},"outputs":[],"source":["import os\n","HOME = os.getcwd()\n","print(HOME)"]},{"cell_type":"markdown","metadata":{"id":"3C3EO_2zNChu"},"source":["## Install YOLOv8\n","\n","⚠️ YOLOv8 is still under heavy development. Breaking changes are being introduced almost weekly. We strive to make our YOLOv8 notebooks work with the latest version of the library. Last tests took place on **27.01.2023** with version **YOLOv8.0.20**.\n","\n","If you notice that our notebook behaves incorrectly - especially if you experience errors that prevent you from going through the tutorial - don't hesitate! Let us know and open an [issue](https://github.com/roboflow/notebooks/issues) on the Roboflow Notebooks repository.\n","\n","YOLOv8 can be installed in two ways - from the source and via pip. This is because it is the first iteration of YOLO to have an official package."]},{"cell_type":"code","execution_count":null,"metadata":{"colab":{"base_uri":"https://localhost:8080/"},"executionInfo":{"elapsed":13719,"status":"ok","timestamp":1680486228593,"user":{"displayName":"Freedom Tech","userId":"03870511654783100627"},"user_tz":-330},"id":"tdSMcABDNKW-","outputId":"ac19ae99-e76c-4e4e-fd4b-d7f2f460490c"},"outputs":[],"source":["# Pip install method (recommended)\n","\n","# !pip install ultralytics==8.0.20\n","\n","from IPython import display\n","display.clear_output()\n","\n","import ultralytics\n","ultralytics.checks()"]},{"cell_type":"code","execution_count":null,"metadata":{"id":"iVvaIYEEPOty"},"outputs":[],"source":["# Git clone method (for development)\n","\n","# %cd {HOME}\n","# !git clone github.com/ultralytics/ultralytics\n","# %cd {HOME}/ultralytics\n","# !pip install -qe ultralytics\n","\n","# from IPython import display\n","# display.clear_output()\n","\n","# import ultralytics\n","# ultralytics.checks()"]},{"cell_type":"code","execution_count":3,"metadata":{"executionInfo":{"elapsed":2,"status":"ok","timestamp":1680486232623,"user":{"displayName":"Freedom Tech","userId":"03870511654783100627"},"user_tz":-330},"id":"VOEYrlBoP9-E"},"outputs":[],"source":["from ultralytics import YOLO\n","\n","from IPython.display import display, Image"]},{"cell_type":"markdown","metadata":{"id":"HnnZSm5OQfPQ"},"source":["## CLI Basics "]},{"cell_type":"markdown","metadata":{"id":"K33S7zlkQku0"},"source":["If you want to train, validate or run inference on models and don't need to make any modifications to the code, using YOLO command line interface is the easiest way to get started. Read more about CLI in [Ultralytics YOLO Docs](https://v8docs.ultralytics.com/cli/).\n","\n","```\n","yolo task=detect    mode=train    model=yolov8n.yaml      args...\n","          classify       predict        yolov8n-cls.yaml  args...\n","          segment        val            yolov8n-seg.yaml  args...\n","                         export         yolov8n.pt        format=onnx  args...\n","```"]},{"cell_type":"markdown","metadata":{"id":"s5RGYA6sPgEd"},"source":["## Inference with Pre-trained COCO Model"]},{"cell_type":"markdown","metadata":{"id":"fT1qD4toTTw0"},"source":["### 💻 CLI"]},{"cell_type":"markdown","metadata":{"id":"ZaE1kLS8R4CV"},"source":["`yolo mode=predict` runs YOLOv8 inference on a variety of sources, downloading models automatically from the latest YOLOv8 release, and saving results to `runs/predict`."]},{"cell_type":"code","execution_count":null,"metadata":{"colab":{"base_uri":"https://localhost:8080/"},"executionInfo":{"elapsed":18219,"status":"ok","timestamp":1680486255682,"user":{"displayName":"Freedom Tech","userId":"03870511654783100627"},"user_tz":-330},"id":"FDbMt_M6PiXb","outputId":"21c73034-8024-4b69-fa5d-31355f7b5385"},"outputs":[],"source":["%cd {HOME} \n","# !yolo task=detect mode=predict model=yolov8n.pt conf=0.25 source='https://media.roboflow.com/notebooks/examples/dog.jpeg' save=True"]},{"cell_type":"code","execution_count":null,"metadata":{"colab":{"base_uri":"https://localhost:8080/"},"executionInfo":{"elapsed":3279,"status":"ok","timestamp":1680486393731,"user":{"displayName":"Freedom Tech","userId":"03870511654783100627"},"user_tz":-330},"id":"Jze57H3ayc6b","outputId":"5e76ec23-0048-4610-aca6-09fb89b8d06d"},"outputs":[],"source":["from google.colab import drive\n","drive.mount('/content/gdrive')\n","# !ln -s /content/gdrive/My\\ Drive/ /mydrive\n","# !ls /mydrive\n","\n"]},{"cell_type":"code","execution_count":null,"metadata":{"colab":{"base_uri":"https://localhost:8080/"},"executionInfo":{"elapsed":3394,"status":"ok","timestamp":1680486423482,"user":{"displayName":"Freedom Tech","userId":"03870511654783100627"},"user_tz":-330},"id":"BSd93ZJzZZKt","outputId":"d168d08c-11d3-455c-b203-2d8d3a2aa647"},"outputs":[],"source":["# !mkdir {HOME}/datasets\n","%cd {HOME}/datasets\n","\n","# !unzip /content/gdrive/MyDrive/freedomtech.zip "]},{"cell_type":"markdown","metadata":{"id":"YUjFBKKqXa-u"},"source":["## Custom Training"]},{"cell_type":"code","execution_count":null,"metadata":{"colab":{"base_uri":"https://localhost:8080/"},"executionInfo":{"elapsed":741711,"status":"ok","timestamp":1680488405529,"user":{"displayName":"Freedom Tech","userId":"03870511654783100627"},"user_tz":-330},"id":"D2YkphuiaE7_","outputId":"4cce49b2-f676-4b3d-cb10-0b6a2de3e476"},"outputs":[],"source":["%cd {HOME}\n","\n","# !yolo task=detect mode=train model=yolov8s.pt data=/content/datasets/freedomtech/data.yaml epochs=100 imgsz=800 plots=True"]}],"metadata":{"accelerator":"GPU","colab":{"provenance":[{"file_id":"1XabZtpjdJayBue3Ogr7yGCDaGCjIk460","timestamp":1680488836995}]},"gpuClass":"standard","kernelspec":{"display_name":"Python 3","name":"python3"},"language_info":{"name":"python"}},"nbformat":4,"nbformat_minor":0}