Tạo layer cho speech bubble mượt viền (2025-07-22 14:21Z)

User

tôi đã tạo được mask, mask này chỉ highlight các speech bubble màu trắng, viền speech bubble còn thô , có cách nào từ file mask, tạo được số lượng các layer của speech bubble (đã làm mượt viền) không ?

Assistant

Bạn muốn từ file mask (mask.png) đã có, tách ra từng layer (mỗi layer là một speech bubble riêng biệt), đồng thời làm mượt viền (smooth contour) cho từng bubble. Đây là một bài toán xử lý ảnh khá phổ biến, có thể giải quyết bằng OpenCV và một số bước sau:

1. Đọc mask và tiền xử lý

Đọc ảnh mask (mask.png), chuyển sang ảnh nhị phân (nếu chưa phải).
Làm mượt viền bằng các phép toán morphology (mở, đóng) hoặc Gaussian Blur + threshold lại.

2. Tìm các vùng liên thông (connected components)

Dùng cv2.connectedComponents hoặc cv2.findContours để tìm từng vùng (mỗi vùng là một speech bubble).

3. Làm mượt viền từng vùng

Với mỗi contour, có thể dùng cv2.approxPolyDP để làm mượt, hoặc dùng cv2.GaussianBlur/cv2.morphologyEx trước khi tìm contour.

4. Tạo layer cho từng bubble

Với mỗi contour, tạo một mask riêng (layer) chỉ chứa vùng đó.

5. Lưu từng layer ra file (hoặc trả về list numpy array)

Ví dụ code (Python, OpenCV)

import cv2
import numpy as np
import os

# Đọc mask
mask = cv2.imread('data/mask.png', cv2.IMREAD_GRAYSCALE)

# Làm mượt viền toàn bộ mask
blurred = cv2.GaussianBlur(mask, (7, 7), 0)
_, binary = cv2.threshold(blurred, 127, 255, cv2.THRESH_BINARY)

# Tìm các vùng liên thông (contour)
contours, _ = cv2.findContours(binary, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

layers = []
output_dir = 'data/speech_bubble_layers'
os.makedirs(output_dir, exist_ok=True)

for i, cnt in enumerate(contours):
    # Làm mượt contour
    epsilon = 0.01 * cv2.arcLength(cnt, True)
    smoothed_cnt = cv2.approxPolyDP(cnt, epsilon, True)
    
    # Tạo mask cho từng bubble
    layer = np.zeros_like(mask)
    cv2.drawContours(layer, [smoothed_cnt], -1, 255, -1)
    layers.append(layer)
    
    # Lưu ra file
    cv2.imwrite(f'{output_dir}/bubble_{i+1}.png', layer)

print(f'Đã tách được {len(layers)} layer speech bubble, lưu tại {output_dir}')

Giải thích:

Làm mượt viền: GaussianBlur + threshold giúp loại bỏ noise nhỏ, viền sẽ mượt hơn. cv2.approxPolyDP giúp làm mượt contour.
Tách layer: Mỗi contour là một bubble, vẽ lên mask trắng đen mới.
Lưu file: Mỗi layer là một file PNG riêng.

Nếu bạn muốn xuất ra dạng numpy array hoặc overlay lên ảnh gốc, có thể mở rộng thêm.

Bạn có muốn mình tạo sẵn script hoàn chỉnh cho bạn không? Hoặc bạn muốn xuất ra định dạng nào khác ngoài PNG?

3.1 KiB Raw Permalink Blame History