Thông báo: Download 4 khóa học Python từ cơ bản đến nâng cao tại đây.
Cách background của hình ảnh trong Python
Trong quá trình xử lý hình ảnh, việc xóa nền là một thao tác phổ biến nhưng không phải lúc nào cũng dễ dàng thực hiện. Tuy nhiên, trong Python, có một công cụ hỗ trợ mã nguồn mở tên là rembg, giúp đơn giản hóa việc xóa nền khỏi hình ảnh chỉ với vài dòng mã. Bài viết này sẽ hướng dẫn bạn cách cài đặt và sử dụng rembg để xử lý nền hình ảnh một cách nhanh chóng và hiệu quả.
Input
Ouput:
Cài đặt rembg trong Python
Trước tiên, hãy cài đặt rembg. Để sử dụng trên CPU, bạn dùng lệnh sau:
Bài viết này được đăng tại [free tuts .net]
pip install rembg
Nếu bạn có GPU, bạn có thể cài đặt phiên bản hỗ trợ GPU:
pip install rembg[gpu]
Mẹo: Để sử dụng GPU miễn phí, bạn có thể dùng Google Colab.
Đảm bảo rằng phiên bản của thư viện Pillow đang dùng là từ 9.3.0 trở lên. Bạn có thể kiểm tra phiên bản hiện tại bằng đoạn mã sau:
import PIL PIL.__version__
Nếu chưa đạt yêu cầu, bạn có thể cài đặt lại Pillow với lệnh này (cách này đặc biệt hữu ích khi sử dụng trên Google Colab):
pip install --ignore-installed Pillow==9.3.0
Xóa nền hình ảnh
Khi mọi thứ đã sẵn sàng, bạn có thể xóa nền của hình ảnh bằng đoạn mã dưới đây:
from rembg import remove input_path = 'input.png' # Đường dẫn đến hình ảnh gốc output_path = 'output.png' # Đường dẫn để lưu ảnh đã xóa nền with open(input_path, 'rb') as i: with open(output_path, 'wb') as o: input = i.read() output = remove(input) o.write(output)
Đoạn mã trên sẽ đọc hình ảnh gốc, áp dụng lệnh xóa nền và lưu kết quả ra một tệp hình ảnh mới.
Sử dụng rembg qua dòng lệnh
Ngoài ra, bạn cũng có thể sử dụng rembg trực tiếp từ dòng lệnh:
rembg i path/to/input.png path/to/output.png
Kết bài
Trên đây là hướng dẫn cơ bản để sử dụng rembg trong Python nhằm xóa nền hình ảnh một cách dễ dàng. Với chỉ vài dòng lệnh đơn giản, bạn có thể xử lý hình ảnh hiệu quả, tiết kiệm thời gian và công sức. Công cụ rembg là một lựa chọn tuyệt vời cho các dự án xử lý hình ảnh từ cơ bản đến nâng cao. Hy vọng qua bài viết này, bạn sẽ áp dụng được rembg vào các ứng dụng của mình một cách thành công và linh hoạt.