输入平面图图像

上面的图像是我输入的平面图,我需要分别识别每个房间,然后裁剪这些房间。之后,我可以将这些图像用于后续步骤。到目前为止,我能够使用 cv2.connectedComponentsWithStats 从输入平面图中删除小项目。所以我认为这将有助于轻松识别墙壁。之后我的输入图像看起来像这样。
去除小物体后的输出图像

然后我做了 MorphologicalTransform 从图像中删除文本和其他符号,只留下墙壁。之后我的输入图像看起来像这样。
形态变换后

所以我能够识别墙壁。然后我如何使用这些墙从原始输入平面图中裁剪房间。有人能帮我吗?您可以在此链接中找到我的 Python 代码。下载我的代码 或
#Import packages
import os
import cv2
import numpy as np
import tensorflow as tf
import sys
# This is needed since the notebook is stored in the object_detection folder.
sys.path.append("..")
# Import utilites
from utils import label_map_util
from utils import visualization_utils as vis_util
# Name of the directory containing the object detection module we're using
MODEL_NAME = 'inference_graph'
IMAGE_NAME = 'floorplan2.jpg'
#Remove Small Items
im_gray = …Run Code Online (Sandbox Code Playgroud) 我使用 AWS Linux 虚拟机作为我的生产服务器。当我尝试从 AWS Linux 虚拟机登录 Docker-hub 时,它显示尝试连接到 docker 守护进程套接字时权限被拒绝。互联网上有一些解决方案,这些解决方案需要“Sudo”访问。但我只是使用 SSH 使用我创建的名为 prod-user 的用户连接到虚拟机。因此我没有密码。