glenn-jocher
commited on
Commit
•
3aeb57d
1
Parent(s):
5ed2860
Created using Colaboratory
Browse files- tutorial.ipynb +7 -24
tutorial.ipynb
CHANGED
@@ -666,31 +666,14 @@
|
|
666 |
"source": [
|
667 |
"# 3. Train\n",
|
668 |
"\n",
|
669 |
-
"
|
670 |
-
|
671 |
-
|
672 |
-
|
673 |
-
|
674 |
-
|
675 |
-
"id": "Knxi2ncxWffW"
|
676 |
-
},
|
677 |
-
"source": [
|
678 |
-
"# Download COCO128\n",
|
679 |
-
"torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/releases/download/v1.0/coco128.zip', 'tmp.zip')\n",
|
680 |
-
"!unzip -q tmp.zip -d ../datasets && rm tmp.zip"
|
681 |
-
],
|
682 |
-
"execution_count": null,
|
683 |
-
"outputs": []
|
684 |
-
},
|
685 |
-
{
|
686 |
-
"cell_type": "markdown",
|
687 |
-
"metadata": {
|
688 |
-
"id": "_pOkGLv1dMqh"
|
689 |
-
},
|
690 |
-
"source": [
|
691 |
-
"Train a YOLOv5s model on [COCO128](https://www.kaggle.com/ultralytics/coco128) with `--data coco128.yaml`, starting from pretrained `--weights yolov5s.pt`, or from randomly initialized `--weights '' --cfg yolov5s.yaml`. Models are downloaded automatically from the [latest YOLOv5 release](https://github.com/ultralytics/yolov5/releases), and **COCO, COCO128, and VOC datasets are downloaded automatically** on first use.\n",
|
692 |
"\n",
|
693 |
-
"
|
694 |
]
|
695 |
},
|
696 |
{
|
|
|
666 |
"source": [
|
667 |
"# 3. Train\n",
|
668 |
"\n",
|
669 |
+
"Train a YOLOv5s model on the [COCO128](https://www.kaggle.com/ultralytics/coco128) dataset with `--data coco128.yaml`, starting from pretrained `--weights yolov5s.pt`, or from randomly initialized `--weights '' --cfg yolov5s.yaml`. \n",
|
670 |
+
"\n",
|
671 |
+
"* **Pretrained [Models](https://github.com/ultralytics/yolov5/tree/master/models)** are downloaded \n",
|
672 |
+
"automatically from the [latest YOLOv5 release](https://github.com/ultralytics/yolov5/releases)\n",
|
673 |
+
"\n",
|
674 |
+
"* **[Datasets](https://github.com/ultralytics/yolov5/tree/master/data)** available for autodownload include: [COCO](https://github.com/ultralytics/yolov5/blob/master/data/coco.yaml), [COCO128](https://github.com/ultralytics/yolov5/blob/master/data/coco128.yaml), [VOC](https://github.com/ultralytics/yolov5/blob/master/data/VOC.yaml), [Argoverse](https://github.com/ultralytics/yolov5/blob/master/data/Argoverse.yaml), [VisDrone](https://github.com/ultralytics/yolov5/blob/master/data/VisDrone.yaml), [GlobalWheat](https://github.com/ultralytics/yolov5/blob/master/data/GlobalWheat2020.yaml), [xView](https://github.com/ultralytics/yolov5/blob/master/data/xView.yaml), [Objects365](https://github.com/ultralytics/yolov5/blob/master/data/Objects365.yaml), [SKU-110K](https://github.com/ultralytics/yolov5/blob/master/data/SKU-110K.yaml).\n",
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
675 |
"\n",
|
676 |
+
"* **Training Results** are saved to `runs/train/` with incrementing run directories, i.e. `runs/train/exp2`, `runs/train/exp3` etc."
|
677 |
]
|
678 |
},
|
679 |
{
|