widzenie-komputerowe-projekt/training/Yolo.ipynb
2023-02-01 18:42:47 +01:00

77 lines
1.5 KiB
Plaintext

{
"cells": [
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"!git clone https://github.com/ultralytics/ultralytics"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"!pip install -e ./ultralytics"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"!unzip 'Aquarium Combined.v2-raw-1024.yolov8.zip' -d /aqua"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"!yolo task=detect \\\n",
" mode=train \\\n",
" model=yolov8s.pt \\\n",
" data=./data.yaml \\\n",
" epochs=100 \\\n",
" imgsz=640"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"!yolo task=detect \\\n",
" mode=val \\\n",
" model=/content/ultralytics/runs/detect/train/weights/best.pt \\\n",
" data=./data.yaml"
]
}
],
"metadata": {
"kernelspec": {
"display_name": "um",
"language": "python",
"name": "python3"
},
"language_info": {
"name": "python",
"version": "3.9.15 | packaged by conda-forge | (main, Nov 22 2022, 08:41:22) [MSC v.1929 64 bit (AMD64)]"
},
"orig_nbformat": 4,
"vscode": {
"interpreter": {
"hash": "876e189cbbe99a9a838ece62aae1013186c4bb7e0254a10cfa2f9b2381853efb"
}
}
},
"nbformat": 4,
"nbformat_minor": 2
}