Commit
Β·
5c32bd3
1
Parent(s):
53ed872
Created using Colaboratory
Browse files- tutorial.ipynb +8 -7
tutorial.ipynb
CHANGED
@@ -523,10 +523,11 @@
|
|
523 |
{
|
524 |
"cell_type": "markdown",
|
525 |
"metadata": {
|
526 |
-
"id": "
|
527 |
},
|
528 |
"source": [
|
529 |
-
"<
|
|
|
530 |
"\n",
|
531 |
"This is the **official YOLOv5 π notebook** authored by **Ultralytics**, and is freely available for redistribution under the [GPL-3.0 license](https://choosealicense.com/licenses/gpl-3.0/). \n",
|
532 |
"For more information please visit https://github.com/ultralytics/yolov5 and https://ultralytics.com. Thank you!"
|
@@ -563,7 +564,7 @@
|
|
563 |
"clear_output()\n",
|
564 |
"print(f\"Setup complete. Using torch {torch.__version__} ({torch.cuda.get_device_properties(0).name if torch.cuda.is_available() else 'CPU'})\")"
|
565 |
],
|
566 |
-
"execution_count":
|
567 |
"outputs": [
|
568 |
{
|
569 |
"output_type": "stream",
|
@@ -681,7 +682,7 @@
|
|
681 |
"torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/releases/download/v1.0/coco2017val.zip', 'tmp.zip')\n",
|
682 |
"!unzip -q tmp.zip -d ../ && rm tmp.zip"
|
683 |
],
|
684 |
-
"execution_count":
|
685 |
"outputs": [
|
686 |
{
|
687 |
"output_type": "display_data",
|
@@ -721,7 +722,7 @@
|
|
721 |
"# Run YOLOv5x on COCO val2017\n",
|
722 |
"!python test.py --weights yolov5x.pt --data coco.yaml --img 640 --iou 0.65 --half"
|
723 |
],
|
724 |
-
"execution_count":
|
725 |
"outputs": [
|
726 |
{
|
727 |
"output_type": "stream",
|
@@ -845,7 +846,7 @@
|
|
845 |
"torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/releases/download/v1.0/coco128.zip', 'tmp.zip')\n",
|
846 |
"!unzip -q tmp.zip -d ../ && rm tmp.zip"
|
847 |
],
|
848 |
-
"execution_count":
|
849 |
"outputs": [
|
850 |
{
|
851 |
"output_type": "display_data",
|
@@ -923,7 +924,7 @@
|
|
923 |
"# Train YOLOv5s on COCO128 for 3 epochs\n",
|
924 |
"!python train.py --img 640 --batch 16 --epochs 3 --data coco128.yaml --weights yolov5s.pt --cache"
|
925 |
],
|
926 |
-
"execution_count":
|
927 |
"outputs": [
|
928 |
{
|
929 |
"output_type": "stream",
|
|
|
523 |
{
|
524 |
"cell_type": "markdown",
|
525 |
"metadata": {
|
526 |
+
"id": "t6MPjfT5NrKQ"
|
527 |
},
|
528 |
"source": [
|
529 |
+
"<a align=\"left\" href=\"https://ultralytics.com/yolov5\" target=\"_blank\">\n",
|
530 |
+
"<img src=\"https://github.com/ultralytics/yolov5/releases/download/v1.0/splash_horizontal.png\"></a>\n",
|
531 |
"\n",
|
532 |
"This is the **official YOLOv5 π notebook** authored by **Ultralytics**, and is freely available for redistribution under the [GPL-3.0 license](https://choosealicense.com/licenses/gpl-3.0/). \n",
|
533 |
"For more information please visit https://github.com/ultralytics/yolov5 and https://ultralytics.com. Thank you!"
|
|
|
564 |
"clear_output()\n",
|
565 |
"print(f\"Setup complete. Using torch {torch.__version__} ({torch.cuda.get_device_properties(0).name if torch.cuda.is_available() else 'CPU'})\")"
|
566 |
],
|
567 |
+
"execution_count": null,
|
568 |
"outputs": [
|
569 |
{
|
570 |
"output_type": "stream",
|
|
|
682 |
"torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/releases/download/v1.0/coco2017val.zip', 'tmp.zip')\n",
|
683 |
"!unzip -q tmp.zip -d ../ && rm tmp.zip"
|
684 |
],
|
685 |
+
"execution_count": null,
|
686 |
"outputs": [
|
687 |
{
|
688 |
"output_type": "display_data",
|
|
|
722 |
"# Run YOLOv5x on COCO val2017\n",
|
723 |
"!python test.py --weights yolov5x.pt --data coco.yaml --img 640 --iou 0.65 --half"
|
724 |
],
|
725 |
+
"execution_count": null,
|
726 |
"outputs": [
|
727 |
{
|
728 |
"output_type": "stream",
|
|
|
846 |
"torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/releases/download/v1.0/coco128.zip', 'tmp.zip')\n",
|
847 |
"!unzip -q tmp.zip -d ../ && rm tmp.zip"
|
848 |
],
|
849 |
+
"execution_count": null,
|
850 |
"outputs": [
|
851 |
{
|
852 |
"output_type": "display_data",
|
|
|
924 |
"# Train YOLOv5s on COCO128 for 3 epochs\n",
|
925 |
"!python train.py --img 640 --batch 16 --epochs 3 --data coco128.yaml --weights yolov5s.pt --cache"
|
926 |
],
|
927 |
+
"execution_count": null,
|
928 |
"outputs": [
|
929 |
{
|
930 |
"output_type": "stream",
|