Share Tools tách design pod

[hathanh34]

Câu chuyện là e đang đi code một con bot để tách design pod thì vô tình tìm được quả này tách ngon phết các bác ạ:

Để xem link bị ẩn này, trước tiên bạn cần có tài khoản, vui lòng Đăng ký hoặc Đăng nhập

 

[cookingtomfoodrecipe]

thanks bác ạ

 

[thanhson5036]

Con này dùng để tách nền ạ?

 

[hathanh34]

@thanhson5036 đúng rồi ạ nhưng mà nó là source. Nguồn nó cũng là thư viện lên phù hợp cho ae nào biết code hoặc dùng chatgpt đc thì sẽ sài đc

 

[hathanh34]

pip install torch
pip install torchvision
pip install opencv-python
pip install numpy
pip install pillow

 

[hathanh34]

import cv2
import numpy as np
import torch
from torchvision import models, transforms
from PIL import Image
# Tải mô hình DeepLabV3+ đã được huấn luyện trước
def load_model():
model = models.segmentation.deeplabv3_resnet101(pretrained=True).eval()
return model
# Hàm xử lý ảnh với DeepLabV3+
def predict(model, image):
preprocess = transforms.Compose([
transforms.ToTensor(),
transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
])
input_tensor = preprocess(image)
input_batch = input_tensor.unsqueeze(0)
with torch.no_grad():
output = model(input_batch)['out'][0]
output_predictions = output.argmax(0).byte().cpu().numpy()
return output_predictions
# Hàm tách nền
def remove_background(image_path, output_path):
# Tải mô hình
model = load_model()
# Đọc ảnh đầu vào
image = Image.open(image_path).convert('RGB')
# Dự đoán mask
mask = predict(model, image)
# Chọn lớp người (class = 15) từ mask
person_class = 15
binary_mask = (mask == person_class).astype(np.uint8) * 255
# Đọc lại ảnh đầu vào với OpenCV
image = cv2.imread(image_path)
# Tạo ảnh kết quả với nền trong suốt
result = cv2.bitwise_and(image, image, mask=binary_mask)
b_channel, g_channel, r_channel = cv2.split(result)
a_channel = binary_mask
result = cv2.merge((b_channel, g_channel, r_channel, a_channel))
# Lưu ảnh kết quả
cv2.imwrite(output_path, result)
# Đường dẫn tới ảnh đầu vào và đầu ra
image_path = "input_image.jpg" # Đường dẫn tới ảnh áo thun cần tách nền
output_path = "output_image.png" # Đường dẫn lưu ảnh kết quả
# Gọi hàm tách nền
remove_background(image_path, output_path)

 

[hathanh34]

có đoạn code tham khảo cho các bác haha

 

[nyattan]

@hathanh34 bác làm video hướng dẫn cài đặt được k ạ

 

[duyen]

thanks bác ạ

 

[o0magiz0o]

có video hướng dẫn ko ạ

 

[thanh169]

bác có thể hướng dẫn sử dụng được ko ạ