From df48c205c5fc7be5af6b067da1f7cb3efb770d88 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 8 Aug 2023 02:13:55 +0200 Subject: [PATCH] Update notebooks `torch.hub.load()` examples (#11952) * Update tutorial.ipynb Signed-off-by: Glenn Jocher * Update tutorial.ipynb Signed-off-by: Glenn Jocher * Update tutorial.ipynb Signed-off-by: Glenn Jocher * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update tutorial.ipynb Signed-off-by: Glenn Jocher --------- Signed-off-by: Glenn Jocher Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> --- classify/tutorial.ipynb | 2 +- segment/tutorial.ipynb | 2 +- tutorial.ipynb | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/classify/tutorial.ipynb b/classify/tutorial.ipynb index 75eebd8e1132..844da0c418f5 100644 --- a/classify/tutorial.ipynb +++ b/classify/tutorial.ipynb @@ -1445,7 +1445,7 @@ "# YOLOv5 PyTorch HUB Inference (DetectionModels only)\n", "import torch\n", "\n", - "model = torch.hub.load('ultralytics/yolov5', 'yolov5s') # yolov5n - yolov5x6 or custom\n", + "model = torch.hub.load('ultralytics/yolov5', 'yolov5s', force_reload=True, trust_repo=True) # or yolov5n - yolov5x6 or custom\n", "im = 'https://ultralytics.com/images/zidane.jpg' # file, Path, PIL.Image, OpenCV, nparray, list\n", "results = model(im) # inference\n", "results.print() # or .show(), .save(), .crop(), .pandas(), etc." diff --git a/segment/tutorial.ipynb b/segment/tutorial.ipynb index 0ece0f60e4d6..cd215e7f3467 100644 --- a/segment/tutorial.ipynb +++ b/segment/tutorial.ipynb @@ -558,7 +558,7 @@ "# YOLOv5 PyTorch HUB Inference (DetectionModels only)\n", "import torch\n", "\n", - "model = torch.hub.load('ultralytics/yolov5', 'yolov5s-seg') # yolov5n - yolov5x6 or custom\n", + "model = torch.hub.load('ultralytics/yolov5', 'yolov5s-seg', force_reload=True, trust_repo=True) # or yolov5n - yolov5x6 or custom\n", "im = 'https://ultralytics.com/images/zidane.jpg' # file, Path, PIL.Image, OpenCV, nparray, list\n", "results = model(im) # inference\n", "results.print() # or .show(), .save(), .crop(), .pandas(), etc." diff --git a/tutorial.ipynb b/tutorial.ipynb index 440e370ce724..f666dbde7e8c 100644 --- a/tutorial.ipynb +++ b/tutorial.ipynb @@ -602,4 +602,4 @@ "outputs": [] } ] -} \ No newline at end of file +}