Проблема PyTorch при попытке загрузить модель YOLOv5 ⇐ Python

[Anonymous]

Я использую OpenCV и модель YOLOv5 как часть более крупного проекта и написал базовый код:

Код: Выделить всё

import torch
import cv2

device = torch.device("cpu")
model = torch.hub.load("ultralytics/yolov5", "yolov5n")
cap = cv2.VideoCapture(0)

while True:
ret, frame = cap.read()
if not ret:
print("Could not access frame!!")
break

results = model(frame)
cv2.imshow('Frame', frame)

if cv2.waitKey(1) & 0xFF == ord('a'):
break

cap.release()
cv2.destroyAllWindows()

Когда я запускаю его, он продолжает выдавать мне эту ошибку:
ImportError: невозможно импортировать имя «tarfile» из «backports» (/opt/ anaconda3/lib/python3.11/site-packages/backports/init.py)
и ошибки возникли из этой строки:

Код: Выделить всё

model = torch.hub.load("ultralytics/yolov5", "yolov5n")

Я искал в Интернете и пробовал другие вещи, но все равно выдавал ту же ошибку.

Подробнее здесь: https://stackoverflow.com/questions/791 ... lov5-model