这是什么意思?列表索引超出范围

问题描述 投票:0回答:0
model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True)
classes = ['bicycle', 'boat', 'bottle', 'bus', 'car', 'cat', 'chair', 'cup', 'dog', 'motorbike', 'people', 'table']
image_path = '/content/drive/MyDrive/phase 2/classwise_excnet_results_5/bicycle/2015_00027.jpg'
image = Image.open(image_path)
results = model(image)
for i, det in enumerate(results.pred):
    print(f'Detections for image {i}:')
    for c in det[:, -1].unique():
        n = (det[:, -1] == c).sum()
        print(f'{n} {classes[int(c)]}(s) detected')
results.render()
plt.imshow(results.render()[0])
plt.show()

我得到的错误是

Using cache found in /root/.cache/torch/hub/ultralytics_yolov5_master
YOLOv5 🚀 2023-4-30 Python-3.10.11 torch-2.0.0+cu118 CUDA:0 (Tesla T4, 15102MiB)

Fusing layers... 
requirements: /root/.cache/torch/hub/requirements.txt not found, check failed.
YOLOv5s summary: 213 layers, 7225885 parameters, 0 gradients
Adding AutoShape... 
Detections for image 0:
3 bicycle(s) detected
2 boat(s) detected
1 bottle(s) detected
1 bus(s) detected
---------------------------------------------------------------------------
IndexError  Traceback (most recent call last)

print(f'{n} {classes[int(c)]}(s) detected')
     
IndexError: list index out of range

我期望从我给定的类中检测到一个对象。但是这个错误被抛出。

python deep-learning computer-vision yolo ipynb
© www.soinside.com 2019 - 2024. All rights reserved.