Блог

Я хочу использовать модель yolov5 в django, но у меня возникли проблемы.

Что я хочу сделать, так это если пользователь загружает изображение на сервер django, обнаружение объекта выполняется с помощью модели yolov5, а затем результат отображается в Интернете. Сам процесс прост. Но я не уверен, как подключить api yolov5 и django.

Как знают те из вас, кто использовал yolo, yolo в основном запускает API на основе команд.

 !python train.py --img 512 --batch 16 --epochs 100 --data ~~   # for training
!python detect.py --weights'/content/yolov5/runs/~~            # for detection
  

Есть ли какой-нибудь способ запустить это в представлении Django?

Я пробовал функции python, такие как execfile() и os.system(), которые выполняют команды с помощью интерпретатора в скрипте Python, но это не сработало должным образом. (Я думаю, что пути между Django и Yolo были скручены.)

На самом деле, если возможно, лучше всего загрузить yolo api как модуль, подобный Keras, и запустить его как функцию, а не как метод command. Но я не могу найти способ использовать yolov5 как модуль или что-то в этом роде.

Как я могу решить эту проблему? Пожалуйста, дайте мне несколько советов.

Обнаружение объектов с помощью Django и Yolov5

Настройте свой проект Django с помощью yolov5:

Я загрузил django-object-detection здесь, для удобства использования.

Models.py:

 import os

from django.db import models
from django.utils.translation import gettext_lazy as _


class ImageModel(models.Model):
    image = models.ImageField(_("image"), upload_to='images')

    class Meta:
        verbose_name = "Image"
        verbose_name_plural = "Images"

    def __str__(self):
        return str(os.path.split(self.image.path)[-1])
  

Views.py:

 import io
from PIL import Image as im
import torch

from django.shortcuts import render
from django.views.generic.edit import CreateView

from .models import ImageModel
from .forms import ImageUploadForm


class UploadImage(CreateView):
    model = ImageModel
    template_name = 'image/imagemodel_form.html'
    fields = ["image"]

    def post(self, request, *args, **kwargs):
        form = ImageUploadForm(request.POST, request.FILES)
        if form.is_valid():
            img = request.FILES.get('image')
            img_instance = ImageModel(
                image=img
            )
            img_instance.save()

            uploaded_img_qs = ImageModel.objects.filter().last()
            img_bytes = uploaded_img_qs.image.read()
            img = im.open(io.BytesIO(img_bytes))

            # Change this to the correct path
            path_hubconfig = "absolute/path/to/yolov5_code"
            path_weightfile = "absolute/path/to/yolov5s.pt"  # or any custom trained model

            model = torch.hub.load(path_hubconfig, 'custom',
                               path=path_weightfile, source='local')

            results = model(img, size=640)
            results.render()
            for img in results.imgs:
                img_base64 = im.fromarray(img)
                img_base64.save("media/yolo_out/image0.jpg", format="JPEG")

            inference_img = "/media/yolo_out/image0.jpg"

            form = ImageUploadForm()
            context = {
                "form": form,
                "inference_img": inference_img
            }
            return render(request, 'image/imagemodel_form.html', context)

        else:
            form = ImageUploadForm()
        context = {
            "form": form
        }
        return render(request, 'image/imagemodel_form.html', context)
  

Многое можно изменить здесь, в этом views.py , например, добавление дополнительных функций и логики, но цель здесь — соединить yolov5 и Django. Конфигурации внутри views.py являются наиболее важными, так как это ворота в yolov5 hubconf.py досье.

Forms.py

 from django import forms
from .models import ImageModel


class ImageUploadForm(forms.ModelForm):
    class Meta:
        model = ImageModel
        fields = ['image']
  

Imagemodel_form.html

 {% extends "base.html" %}
{% load crispy_forms_tags %}


{% block leftbar %}
    <div class="col-sm-3">

    </div>
{% endblock leftbar %} 


{% block content %}
    <div class="col-sm-9">
        <div id="uploadedImage"></div>
            <div class="mt-4">
                <form action="" enctype="multipart/form-data" id="imageUploadForm" method="post"> 
                    {% csrf_token %}
                    {{ form|crispy }}
                    <button class="btn btn-outline-success" type="submit">Submit</button>
                </form>
            </div>
        </div>
        <div class="mt-4">
            {% if inference_img %}
                <img src="{{inference_img}}" class="img-fluid" />
            {% else %}
                The inferenced image will be displayed here.
            {% endif %}
        </div>    
    </div>
{% endblock content %}
  

Изначально простая веб-страница:

После обнаружения: