torch hub工具的使用：torch.hub.load、pytorch预训练模型加载

想用torch hub工具工具调用github里的yolov5模型

地址: https://github.com/ultralytics/yolov5

在使用工具加载模型前需要安装必备的依赖包

pip install tqdm boto3 requests regex sentencepiece sacremoses

import torch

# 模型
model = torch.hub.load('ultralytics/yolov5', 'yolov5s')  # or yolov5n - yolov5x6, custom

# 图像
img = 'https://ultralytics.com/images/zidane.jpg'  # or file, Path, PIL, OpenCV, numpy, list

# 推理
results = model(img)

# 结果
print(results.print())  # or .show(), .save(), .crop(), .pandas(), etc.

就可以调用库了