Как создать 3D-сетку из изображения 2D-маски ⇐ Python

[Anonymous]

Я обучил модель сегментации и обнаружил границы плана этажа. Теперь я хотел создать 3D-сетку этой маски и построить стены в формате, который можно легко показать в Интернете. Это маска, созданная с помощью ai
Это стены< /p>
Теперь я хочу выдавить стены до определенной высоты и создать 3D-модель.
Вот код для извлечения стен. Следующим шагом будет создайте из него 3D-модель. Как я могу это сделать?
import cv2
import os
import numpy as np
import matplotlib.pyplot as plt

# Parameters
wall_height = 150
img_path = 'output_mask.png'

# Load the image
img = cv2.imread(img_path)

# Get the original dimensions of the image
original_height, original_width = img.shape[:2]

# Resize the image while preserving aspect ratio
new_width = 1024
new_height = int((new_width / original_width) * original_height)
img_resized = cv2.resize(img, (new_width, new_height))

# Create a blank mask for the walls
walls = np.zeros(img_resized.shape[:2], dtype=np.uint8)

# Extract the walls (walls are white in the image)
walls[(img_resized == [255, 255, 255]).all(axis=2)] = 255


Подробнее здесь: https://stackoverflow.com/questions/792 ... mask-image