glenn-jocher commited on
Commit
3aeb57d
1 Parent(s): 5ed2860

Created using Colaboratory

Browse files
Files changed (1) hide show
  1. tutorial.ipynb +7 -24
tutorial.ipynb CHANGED
@@ -666,31 +666,14 @@
666
  "source": [
667
  "# 3. Train\n",
668
  "\n",
669
- "Download [COCO128](https://www.kaggle.com/ultralytics/coco128), a small 128-image tutorial dataset, start tensorboard and train YOLOv5s from a pretrained checkpoint for 3 epochs (note actual training is typically much longer, around **300-1000 epochs**, depending on your dataset)."
670
- ]
671
- },
672
- {
673
- "cell_type": "code",
674
- "metadata": {
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
- "All training results are saved to `runs/train/` with incrementing run directories, i.e. `runs/train/exp2`, `runs/train/exp3` etc.\n"
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
  {