Как сконвертировать модель .pt (PyTorch) в torchscript.pt для работы в DJL?
Я пытаюсь воспользоваться моделью YOLOv5l.pt для распознавания объектов на экране компьютера. DJL не принимает: ai.djl.engine.EngineException: PytorchStreamReader failed locating file constants.pkl: file not found
. Прочитал, что надо сконвертировать модель в формат .torchscript.pt. Попытался:
import torch
import sys
sys.path.append("C:/Users/user/Downloads/Yolo/yolov5")
from models.common import DetectMultiBackend
model = DetectMultiBackend("...")
example_input = torch.rand(1, 3, 640, 640)
traced_model = torch.jit.trace(model, example_input)
traced_model.save("...")
Не получилось:
RuntimeError: Tracer cannot infer type of [tensor([[[6.63067e+00, 4.48799e+00, 1.29999e+01, ..., 1.77943e-03, 8.68350e-04, 2.78321e-03], # Прочие веса
Попытался воспользоваться файлом export.py:
python export.py --weights C:/Users/user/AppData/Local/AZComplex/Clicker/plugins/Smart Searcher/resources/yolov5l.pt --include torchscript
Тоже ошибка:
Permission denied: 'C:\\Users\\user\\AppData\\Local\\Programs\\Python\\Python312\\Lib\\site-packages\\mpmath\\__init__.py'
Как избежать ошибок и использовать DJL для работы с YOLO?