glenn-jocher commited on
Commit
1b100cd
·
1 Parent(s): 866bc7d

Created using Colaboratory (#2649)

Browse files
Files changed (1) hide show
  1. tutorial.ipynb +7 -7
tutorial.ipynb CHANGED
@@ -563,7 +563,7 @@
563
  "clear_output()\n",
564
  "print('Setup complete. Using torch %s %s' % (torch.__version__, torch.cuda.get_device_properties(0) if torch.cuda.is_available() else 'CPU'))"
565
  ],
566
- "execution_count": 1,
567
  "outputs": [
568
  {
569
  "output_type": "stream",
@@ -689,7 +689,7 @@
689
  "torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/releases/download/v1.0/coco2017val.zip', 'tmp.zip')\n",
690
  "!unzip -q tmp.zip -d ../ && rm tmp.zip"
691
  ],
692
- "execution_count": 2,
693
  "outputs": [
694
  {
695
  "output_type": "display_data",
@@ -729,7 +729,7 @@
729
  "# Run YOLOv5x on COCO val2017\n",
730
  "!python test.py --weights yolov5x.pt --data coco.yaml --img 640 --iou 0.65"
731
  ],
732
- "execution_count": 3,
733
  "outputs": [
734
  {
735
  "output_type": "stream",
@@ -798,9 +798,9 @@
798
  "source": [
799
  "# Download COCO test-dev2017\n",
800
  "torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/releases/download/v1.0/coco2017labels.zip', 'tmp.zip')\n",
801
- "!unzip -q tmp.zip -d ../ && rm tmp.zip # unzip labels\n",
802
  "!f=\"test2017.zip\" && curl http://images.cocodataset.org/zips/$f -o $f && unzip -q $f && rm $f # 7GB, 41k images\n",
803
- "%mv ./test2017 ./coco/images && mv ./coco ../ # move images to /coco and move /coco next to /yolov5"
804
  ],
805
  "execution_count": null,
806
  "outputs": []
@@ -853,7 +853,7 @@
853
  "torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/releases/download/v1.0/coco128.zip', 'tmp.zip')\n",
854
  "!unzip -q tmp.zip -d ../ && rm tmp.zip"
855
  ],
856
- "execution_count": 4,
857
  "outputs": [
858
  {
859
  "output_type": "display_data",
@@ -930,7 +930,7 @@
930
  "# Train YOLOv5s on COCO128 for 3 epochs\n",
931
  "!python train.py --img 640 --batch 16 --epochs 3 --data coco128.yaml --weights yolov5s.pt --nosave --cache"
932
  ],
933
- "execution_count": 5,
934
  "outputs": [
935
  {
936
  "output_type": "stream",
 
563
  "clear_output()\n",
564
  "print('Setup complete. Using torch %s %s' % (torch.__version__, torch.cuda.get_device_properties(0) if torch.cuda.is_available() else 'CPU'))"
565
  ],
566
+ "execution_count": null,
567
  "outputs": [
568
  {
569
  "output_type": "stream",
 
689
  "torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/releases/download/v1.0/coco2017val.zip', 'tmp.zip')\n",
690
  "!unzip -q tmp.zip -d ../ && rm tmp.zip"
691
  ],
692
+ "execution_count": null,
693
  "outputs": [
694
  {
695
  "output_type": "display_data",
 
729
  "# Run YOLOv5x on COCO val2017\n",
730
  "!python test.py --weights yolov5x.pt --data coco.yaml --img 640 --iou 0.65"
731
  ],
732
+ "execution_count": null,
733
  "outputs": [
734
  {
735
  "output_type": "stream",
 
798
  "source": [
799
  "# Download COCO test-dev2017\n",
800
  "torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/releases/download/v1.0/coco2017labels.zip', 'tmp.zip')\n",
801
+ "!unzip -q tmp.zip -d ../ && rm tmp.zip # unzip labels\n",
802
  "!f=\"test2017.zip\" && curl http://images.cocodataset.org/zips/$f -o $f && unzip -q $f && rm $f # 7GB, 41k images\n",
803
+ "%mv ./test2017 ../coco/images # move to /coco"
804
  ],
805
  "execution_count": null,
806
  "outputs": []
 
853
  "torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/releases/download/v1.0/coco128.zip', 'tmp.zip')\n",
854
  "!unzip -q tmp.zip -d ../ && rm tmp.zip"
855
  ],
856
+ "execution_count": null,
857
  "outputs": [
858
  {
859
  "output_type": "display_data",
 
930
  "# Train YOLOv5s on COCO128 for 3 epochs\n",
931
  "!python train.py --img 640 --batch 16 --epochs 3 --data coco128.yaml --weights yolov5s.pt --nosave --cache"
932
  ],
933
+ "execution_count": null,
934
  "outputs": [
935
  {
936
  "output_type": "stream",