将 numpy 数组保存到图像会将图像部分缩小到错误的大小

Question

我在弄清楚如何在 Blender 脚本中导出外部图像时遇到了这个问题。但我想这不再直接与搅拌机相关，更多的是与 numpy 以及如何处理数组有关。 Here is post about first problem.

所以问题是，当保存 numpy 数组到图像时，它会扭曲并且有多个相同的图像。查看下图以获得更好的理解。

我们的目标是试图弄清楚如何使用 numpy 和 python 使用 blender 自己的像素数据来完成这项工作。因此，避免使用 blender python.

中未包含的 PIL 或 cv2 等库

保存数据时，所有最终尺寸的图像都能正常工作。当尝试将 4 个较小的部分合并为最终的较大图像时，它无法正确导出。

我在搅拌机中用 python 完成了示例脚本来演示问题：

# Example script to show how to merge external images in Blender
# using numpy. In this example we use 4 images (2x2) that should
# be merged to one actual final image. 
# Regular (not cropped render borders) seems to work fine but
# how to merge cropped images properly???
#
# Usage: Just run script and it will export image named "MERGED_IMAGE"
# to root of this project folder and you'll see what's the problem.

import bpy, os
import numpy as np

ctx = bpy.context
scn = ctx.scene

print('START')

# Get all image files
def get_files_in_folder(path):
    path = bpy.path.abspath(path)
    render_files = []
    for root, dirs, files in os.walk(path):
        for file in files:
            if (file.lower().endswith(('.png', '.jpg', '.jpeg', '.tiff', '.bmp', '.gif'))):
                render_files.append(file)
    return render_files

def merge_images(image_files, image_cropped = True):

    image_pixels = []
    final_image_pixels = 0

    print(image_files)

    for file in image_files:
        if image_cropped is True:
            filepath = bpy.path.abspath('//Cropped\' + file)
        else:
            filepath = bpy.path.abspath('//Regular\' + file)
        loaded_pixels = bpy.data.images.load(filepath, check_existing=True).pixels
        image_pixels.append(loaded_pixels)

    np_array = np.array(image_pixels)

    # Merge images
    if image_cropped:
        final_image_pixels = np_array
        # HOW MERGE PROPERLY WHEN USING CROPPED IMAGES???
    else:
        for arr in np_array:
            final_image_pixels += arr

    # Save output image
    output_image = bpy.data.images.new('MERGED_IMAGE', alpha=True, width=256, height=256)
    output_image.file_format = 'PNG'
    output_image.alpha_mode = 'STRAIGHT'
    output_image.pixels = final_image_pixels.ravel()
    output_image.filepath_raw = bpy.path.abspath("//MERGED_IMAGE.png")
    output_image.save()   

images_cropped = get_files_in_folder("//Cropped")
images_regular = get_files_in_folder('//Regular')

# Change between these to get different example
merge_images(images_cropped)
#merge_images(images_regular, False)

print('END')

所以我猜这个问题与如何使用 numpy 处理图像像素数据和数组有关。

这是 zip 文件中的项目文件夹，其中包含工作测试脚本示例，您可以在其中测试它在 blender 中的工作方式。 https://drive.google.com/file/d/1R4G_fubEzFWbHZMLtAAES-QsRhKyLKWb/view?usp=sharing

Answer 1

由于您所有的图像都是 128x128 的相同维度，并且由于 OpenCV 图像是 Numpy 数组，因此这里提供三种方法。您可以使用 cv2.imwrite.

保存图像

输入图像：

方法一： np.hstack + np.vstack

hstack1 = np.hstack((image1, image2))
hstack2 = np.hstack((image3, image4))
hstack_result = np.vstack((hstack1, hstack2))

方法二： np.concatenate

concatenate1 = np.concatenate((image1, image2), axis=1)
concatenate2 = np.concatenate((image3, image4), axis=1)
concatenate_result = np.concatenate((concatenate1, concatenate2), axis=0)

方法三： cv2.hconcat + cv2.vconcat

hconcat1 = cv2.hconcat([image1, image2])
hconcat2 = cv2.hconcat([image3, image4])
hconcat_result = cv2.vconcat([hconcat1, hconcat2])

所有方法的结果应该相同

完整代码

import cv2
import numpy as np

# Load images
image1 = cv2.imread('Fart_1_2.png')
image2 = cv2.imread('Fart_2_2.png')
image3 = cv2.imread('Fart_1_1.png')
image4 = cv2.imread('Fart_2_1.png')

# Method #1
hstack1 = np.hstack((image1, image2))
hstack2 = np.hstack((image3, image4))
hstack_result = np.vstack((hstack1, hstack2))

# Method #2
concatenate1 = np.concatenate((image1, image2), axis=1)
concatenate2 = np.concatenate((image3, image4), axis=1)
concatenate_result = np.concatenate((concatenate1, concatenate2), axis=0) 

# Method #3
hconcat1 = cv2.hconcat([image1, image2])
hconcat2 = cv2.hconcat([image3, image4])
hconcat_result = cv2.vconcat([hconcat1, hconcat2])

# Display
cv2.imshow('concatenate_result', concatenate_result)
cv2.imshow('hstack_result', hstack_result)
cv2.imshow('hconcat_result', hconcat_result)
cv2.waitKey()

将 numpy 数组保存到图像会将图像部分缩小到错误的大小

Saving numpy array to image shrinks image parts to wrong size

python

arrays

numpy

image-processing

blender