소스 검색

Update README.md

modifyDataloader
Glenn Jocher GitHub 3 년 전
부모
커밋
e5d71223b8
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1개의 변경된 파일4개의 추가작업 그리고 5개의 파일을 삭제
  1. +4
    -5
      README.md

+ 4
- 5
README.md 파일 보기

@@ -121,19 +121,18 @@ Done. (0.103s)

### PyTorch Hub

To run **batched inference** with YOLOv5 and [PyTorch Hub](https://github.com/ultralytics/yolov5/issues/36):
Inference with YOLOv5 and [PyTorch Hub](https://github.com/ultralytics/yolov5/issues/36):
```python
import torch

# Model
model = torch.hub.load('ultralytics/yolov5', 'yolov5s')

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

# Inference
results = model(imgs)
results = model(img)
results.print() # or .show(), .save()
```


Loading…
취소
저장