将程序放在yolov5目录下面
import torch
model = torch.hub.load(r’D:\fulin\code\count_car_flow\yolov5_v6.0_20220617/‘, ‘custom’, path=r’best.pt’,source=‘local’)
#model_weights = torch.load(‘yolov5s.pt’) # 替换为你的.pt文件路径
#model.load_state_dict(model_weights)
打印类别
print(model.names)