- 目标
介绍一个用于语义分割后处理和可视化的Python脚本。该脚本通过OpenCV库实现,可以在图像中标记出分割区域的轮廓、中心点和重心,以提供更直观的视觉反馈
import cv2
import numpy as np
filename = './mask/3.png'
# 读取灰度图像
gray_image = cv2.imread(filename, cv2.IMREAD_GRAYSCALE)
# 使用适当的阈值将图像转换为二值图像
_, binary_image = cv2.threshold(gray_image, 128, 255, cv2.THRESH_BINARY)
# 寻找分割区域的轮廓
contours, _ = cv2.findContours(binary_image, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
# 检查是否找到至少一个有效轮廓
if contours:
# 创建与输入图像相同大小