Skip to content

Commit 69eecec

Browse files
authored
1 parent 11c554c commit 69eecec

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

README.md

+3-4
Original file line numberDiff line numberDiff line change
@@ -120,12 +120,11 @@ import torch
120120
# Model
121121
model = torch.hub.load('ultralytics/yolov3', 'yolov3') # or 'yolov3_spp', 'yolov3_tiny'
122122

123-
# Images
124-
dir = 'https://github.com/ultralytics/yolov3/raw/master/data/images/'
125-
imgs = [dir + f for f in ('zidane.jpg', 'bus.jpg')] # batch of images
123+
# Image
124+
img = 'https://ultralytics.com/images/zidane.jpg'
126125

127126
# Inference
128-
results = model(imgs)
127+
results = model(img)
129128
results.print() # or .show(), .save()
130129
```
131130

0 commit comments

Comments
 (0)