Commit
·
1b100cd
1
Parent(s):
866bc7d
Created using Colaboratory (#2649)
Browse files- 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":
|
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":
|
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":
|
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
|
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
|
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":
|
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":
|
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",
|