python - 如何将 cv2 处理后的图像保存到 Django 模型中？

Question

我使用 opencv 裁剪图像，我想将它们保存到模型中，我将文件直接加载到 computeLogoFromMemoryFILE 进行处理，从那里如何将图像保存到 TempImage 模型？

views.py:

form = myForm(request.FILES)
if form.is_valid():
    cropped_image = computeLogoFromMemoryFILE(request.FILES.get('logo'))
    # ...
    temp_image = TempImage.objects.create(image=?)

CV2:

# (np == numpy)
def computeLogoFromMemoryFILE(logo):
    logo.seek(0)
    image = cv2.imdecode(np.fromstring(logo.read(), np.uint8), cv2.IMREAD_UNCHANGED)
    cropped_img = crop_image(image)

---

cropped_image 变量是一个 opencv 数组:

array([[ 52, 218, 255],
    [ 52, 218, 255],
    [ 52, 218, 255],
    ...,
    [ 52, 218, 255],
    [ 52, 218, 255],
    [ 52, 218, 255]]...], dtype=uint8)

我应该如何进行？

Answer 1

在 Django 中，每当你需要操作上传的文件、相似的图像并将它们设置为模型字段时，你必须使用 Django File类，你可以做类似的代码:

from django.core.files import File

def my_view(request):
    ...
    form = myForm(request.FILES)

    if form.is_valid():
        temp_image = myForm.save(commit=False)
        cropped_image = computeLogoFromMemoryFILE(request.FILES.get('logo'))
        with open('path/of/cropped_image.png', 'rb') as destination_file:
            temp_image.image.save('dest.png', File(destination_file), save=False)
        temp_image.save()
    ...

注意:在设置文件到模型字段后，此文件克隆到 MEDIA_ROOT , 最好删除旧图像或使用 BytesIO而不是使用文件来存储处理后的图像。