Created using Colaboratory

This commit is contained in:
Glenn Jocher 2021-05-12 14:28:11 +02:00
parent 331df67aac
commit af7b923bfa

13
tutorial.ipynb vendored
View File

@ -563,7 +563,7 @@
"clear_output()\n",
"print(f\"Setup complete. Using torch {torch.__version__} ({torch.cuda.get_device_properties(0).name if torch.cuda.is_available() else 'CPU'})\")"
],
"execution_count": 1,
"execution_count": null,
"outputs": [
{
"output_type": "stream",
@ -601,7 +601,7 @@
"!python detect.py --weights yolov3.pt --img 640 --conf 0.25 --source data/images/\n",
"Image(filename='runs/detect/exp/zidane.jpg', width=600)"
],
"execution_count": 5,
"execution_count": null,
"outputs": [
{
"output_type": "stream",
@ -681,7 +681,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": 3,
"execution_count": null,
"outputs": [
{
"output_type": "display_data",
@ -721,7 +721,7 @@
"# Run YOLOv3 on COCO val2017\n",
"!python test.py --weights yolov3.pt --data coco.yaml --img 640 --iou 0.65"
],
"execution_count": 4,
"execution_count": null,
"outputs": [
{
"output_type": "stream",
@ -901,7 +901,8 @@
"source": [
"# Weights & Biases (optional)\n",
"%pip install -q wandb\n",
"!wandb login # use 'wandb disabled' or 'wandb enabled' to disable or enable"
"import wandb\n",
"wandb.login()"
],
"execution_count": null,
"outputs": []
@ -919,7 +920,7 @@
"# Train YOLOv3 on COCO128 for 3 epochs\n",
"!python train.py --img 640 --batch 16 --epochs 3 --data coco128.yaml --weights yolov3.pt --nosave --cache"
],
"execution_count": 6,
"execution_count": null,
"outputs": [
{
"output_type": "stream",