|
|
@@ -563,7 +563,7 @@ |
|
|
|
"clear_output()\n", |
|
|
|
"print('Setup complete. Using torch %s %s' % (torch.__version__, torch.cuda.get_device_properties(0) if torch.cuda.is_available() else 'CPU'))" |
|
|
|
], |
|
|
|
"execution_count": 1, |
|
|
|
"execution_count": null, |
|
|
|
"outputs": [ |
|
|
|
{ |
|
|
|
"output_type": "stream", |
|
|
@@ -689,7 +689,7 @@ |
|
|
|
"torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/releases/download/v1.0/coco2017val.zip', 'tmp.zip')\n", |
|
|
|
"!unzip -q tmp.zip -d ../ && rm tmp.zip" |
|
|
|
], |
|
|
|
"execution_count": 2, |
|
|
|
"execution_count": null, |
|
|
|
"outputs": [ |
|
|
|
{ |
|
|
|
"output_type": "display_data", |
|
|
@@ -729,7 +729,7 @@ |
|
|
|
"# Run YOLOv5x on COCO val2017\n", |
|
|
|
"!python test.py --weights yolov5x.pt --data coco.yaml --img 640 --iou 0.65" |
|
|
|
], |
|
|
|
"execution_count": 3, |
|
|
|
"execution_count": null, |
|
|
|
"outputs": [ |
|
|
|
{ |
|
|
|
"output_type": "stream", |
|
|
@@ -798,9 +798,9 @@ |
|
|
|
"source": [ |
|
|
|
"# Download COCO test-dev2017\n", |
|
|
|
"torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/releases/download/v1.0/coco2017labels.zip', 'tmp.zip')\n", |
|
|
|
"!unzip -q tmp.zip -d ../ && rm tmp.zip # unzip labels\n", |
|
|
|
"!unzip -q tmp.zip -d ../ && rm tmp.zip # unzip labels\n", |
|
|
|
"!f=\"test2017.zip\" && curl http://images.cocodataset.org/zips/$f -o $f && unzip -q $f && rm $f # 7GB, 41k images\n", |
|
|
|
"%mv ./test2017 ./coco/images && mv ./coco ../ # move images to /coco and move /coco next to /yolov5" |
|
|
|
"%mv ./test2017 ../coco/images # move to /coco" |
|
|
|
], |
|
|
|
"execution_count": null, |
|
|
|
"outputs": [] |
|
|
@@ -853,7 +853,7 @@ |
|
|
|
"torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/releases/download/v1.0/coco128.zip', 'tmp.zip')\n", |
|
|
|
"!unzip -q tmp.zip -d ../ && rm tmp.zip" |
|
|
|
], |
|
|
|
"execution_count": 4, |
|
|
|
"execution_count": null, |
|
|
|
"outputs": [ |
|
|
|
{ |
|
|
|
"output_type": "display_data", |
|
|
@@ -930,7 +930,7 @@ |
|
|
|
"# Train YOLOv5s on COCO128 for 3 epochs\n", |
|
|
|
"!python train.py --img 640 --batch 16 --epochs 3 --data coco128.yaml --weights yolov5s.pt --nosave --cache" |
|
|
|
], |
|
|
|
"execution_count": 5, |
|
|
|
"execution_count": null, |
|
|
|
"outputs": [ |
|
|
|
{ |
|
|
|
"output_type": "stream", |