apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Created using Colaboratory

Browse Source
This commit is contained in:
Glenn Jocher 2021-05-12 14:28:11 +02:00
parent 331df67aac
commit af7b923bfa
1 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tutorial.ipynb vendored
View File

@ -563,7 +563,7 @@
"clear_output()\n", "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'})\")" "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": [ "outputs": [
{ {
"output_type": "stream", "output_type": "stream",
@ -601,7 +601,7 @@
"!python detect.py --weights yolov3.pt --img 640 --conf 0.25 --source data/images/\n", "!python detect.py --weights yolov3.pt --img 640 --conf 0.25 --source data/images/\n",
"Image(filename='runs/detect/exp/zidane.jpg', width=600)" "Image(filename='runs/detect/exp/zidane.jpg', width=600)"
], ],
"execution_count": 5, "execution_count": null,
"outputs": [ "outputs": [
{ {
"output_type": "stream", "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", "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" "!unzip -q tmp.zip -d ../ && rm tmp.zip"
], ],
"execution_count": 3, "execution_count": null,
"outputs": [ "outputs": [
{ {
"output_type": "display_data", "output_type": "display_data",
@ -721,7 +721,7 @@
"# Run YOLOv3 on COCO val2017\n", "# Run YOLOv3 on COCO val2017\n",
"!python test.py --weights yolov3.pt --data coco.yaml --img 640 --iou 0.65" "!python test.py --weights yolov3.pt --data coco.yaml --img 640 --iou 0.65"
], ],
"execution_count": 4, "execution_count": null,
"outputs": [ "outputs": [
{ {
"output_type": "stream", "output_type": "stream",
@ -901,7 +901,8 @@
"source": [ "source": [
"# Weights & Biases (optional)\n", "# Weights & Biases (optional)\n",
"%pip install -q wandb\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, "execution_count": null,
"outputs": [] "outputs": []
@ -919,7 +920,7 @@
"# Train YOLOv3 on COCO128 for 3 epochs\n", "# 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" "!python train.py --img 640 --batch 16 --epochs 3 --data coco128.yaml --weights yolov3.pt --nosave --cache"
], ],
"execution_count": 6, "execution_count": null,
"outputs": [ "outputs": [
{ {
"output_type": "stream", "output_type": "stream",