Как интегрировать ансамблевую модель PyTorch YOLO (файл .pt) в веб-приложение Django для прогнозирования изображений в реальном времени?
Я создаю веб-приложение Django, в которое пользователи могут загружать изображения для обнаружения объектов, используя модель YOLO ensemble, сохраненную в виде файла .pt. Эта модель включает в себя такие ключи, как 'model', 'names', 'stride' и 'ensemble_info'. Пока что я могу загрузить модель с помощью torch.load(), но я не уверен, как это сделать:
Предварительно обработать загруженное изображение в режиме просмотра Django,
Выполните логический вывод, используя загруженную ансамблевую модель, и
Возвращает результаты в виде ограничивающей рамки во внешний интерфейс (в идеале в виде JSON или нарисованного изображения).
Окружение: Python 3.10, Django 4.x, Torch 1.13+
Кто-нибудь делал что-то подобное, или вы можете указать на пример репозитория / руководства? Любая помощь или фрагменты кода были бы признательны!