Где найти адекватную документацию к torch?

Question

[ADKX]

все найденные мною гайды показывают только вывод квадратика с классом, я так и не смог найти документацию в которой бы говорилось как мне получение название классов, координаты и точность найденных объектов для работы с ними в кодe.

import torch
import cv2
import numpy as np

model = torch.hub.load('ultralytics/yolov5', 'custom', path='yolov5/runs/train/exp5/weights/best.pt', force_reload=True)

cap = cv2.VideoCapture(0)
while cap.isOpened():
    ret, frame = cap.read()
    
    results = model(frame)
    
    cv2.imshow('YOLO', np.squeeze(results.render()))
    
    if cv2.waitKey(10) & 0xFF == ord('q'):
        break
cap.release()
cv2.destroyAllWindows()

подскажите, что возвращает "results = model(frame)" или ссылки где можно посмотреть примеры получения нужных мне значений.

Comments

[Vindicar]

Ну для начала можешь посмотреть, что ещё лежит в объекте results, помимо метода render.
Вообще в Питоне отличная рефлексия "из коробки", для любого объекта можно узнать доступные методы и свойства, а то и какую-никакую справку получить. dir() и help() тебе в руки.