Cách xóa nền khỏi hình ảnh bằng Python

Trong hướng dẫn này, chúng tôi sẽ sử dụng hai mô-đun Python để tạo một ứng dụng GUI sẽ xóa nền khỏi hình ảnh. Mô-đun đầu tiên, rembg từ Daniel Gatis sẽ xóa nền khỏi bất kỳ hình ảnh nào được hiển thị cho nó. Mô-đun thứ hai, easygui cung cấp phương tiện để tạo hộp thoại và menu bằng bộ công cụ của hệ điều hành. Vì vậy, hộp thoại mở/lưu tệp sẽ trông giống hệt hộp thoại được sử dụng trong nhiều ứng dụng khác

Để tiếp tục dự án này, bạn có thể tạo các ứng dụng của mình bằng cách đóng gói mã dự án thành một tệp thực thi duy nhất

Đang cài đặt

Trước khi viết một dòng mã, chúng ta cần sắp xếp mọi thứ theo thứ tự. Đầu tiên chúng ta sẽ tạo một thư mục để lưu trữ những hình ảnh mà chúng ta sẽ làm việc với. Sau đó, chúng tôi sẽ mở trình soạn thảo Python cho phần viết mã của dự án

1. Tạo một thư mục mới trên màn hình của bạn có tên là rembg

2. Trong thư mục, đặt một hình ảnh mà bạn muốn xóa nền khỏi

3. Mở trình chỉnh sửa Python ưa thích của bạn, chúng tôi thích Thonny hơn vì nó cung cấp giao diện người dùng đơn giản. Làm theo hướng dẫn này để cài đặt Thonny

Cài đặt các mô-đun Python

Để sử dụng Rembg, trước tiên chúng ta cần tải xuống và cài đặt mô-đun Python của nó. Điều này có thể được xử lý thông qua trình quản lý gói tích hợp của Thonny hoặc thông qua công cụ đóng gói của Python, pip.  

Cài đặt qua Thonny

1. Nhấp vào Công cụ >> Quản lý gói

2. Tìm kiếm rembg và nhấp vào Tìm kiếm trên PyPi. Chọn rembg từ danh sách trả về

3. Nhấp vào Cài đặt để tải xuống và cài đặt rembg cho Python. Chúng tôi đã cài đặt rembg trên hệ thống của mình, do đó có nút gỡ cài đặt

4. Lặp lại bước 2 và 3, nhưng lần này tìm kiếm và cài đặt easygui

Cài đặt qua Pip

Nếu bạn đang sử dụng một trình soạn thảo Python khác, bạn sẽ cần cài đặt các gói Python bằng cách sử dụng pip

1. Mở Command Prompt và cài đặt rembg. Nhấn Enter để bắt đầu quá trình

pip install rembg

2. Cài đặt easygui bằng pip. Easygui cung cấp giao diện người dùng cơ bản cho các thao tác mở và lưu tệp

pip install easygui

Viết mã

Code về cơ bản rất đơn giản, chỉ với 8 dòng Python là chúng ta có thể xóa nền khỏi bất kỳ hình ảnh nào. Các mô-đun cơ bản, rembg và easygui sẽ thực hiện tất cả các công việc nặng nhọc cho chúng tôi

1. Từ mô-đun rembg nhập lớp xóa. Đây là những gì chúng ta sẽ sử dụng để loại bỏ nền

from rembg import remove

2. Từ Thư viện hình ảnh Python (PIL), nhập Hình ảnh. PIL là một mô-đun mạnh mẽ chứa nhiều tùy chọn khác nhau để tạo và làm việc với hình ảnh và luồng hình ảnh

from PIL import Image

3. Nhập mô-đun easygui và tạo một tham chiếu đến nó dưới dạng “ví dụ”. Easygui là GUI của chúng tôi dành cho các thao tác tệp cơ bản. Đổi tên nó thành “ví dụ” giúp làm việc dễ dàng hơn

import easygui as eg

4. Tạo một đối tượng, input_path và sử dụng nó để lưu trữ đường dẫn và tên của tệp mà chúng tôi muốn xóa nền khỏi. Sử dụng hộp thoại mở tệp của easygui, chúng tôi đặt tiêu đề cho hộp thoại, để giải thích nó dùng để làm gì. Tệp đã chọn và đường dẫn của nó được lưu vào đối tượng input_path

input_path = eg.fileopenbox(title='Select image file')

5. Tạo một đối tượng, output_path và sử dụng hộp thoại lưu tệp của easygui để nắm bắt đường dẫn tệp và lưu nó vào đối tượng

from PIL import Image

6. Tạo một đối tượng, nhập và sử dụng nó để mở và lưu trữ hình ảnh qua PIL's Image. chức năng mở

from PIL import Image

7. Sử dụng rembg để xóa nền khỏi hình ảnh

output = remove(input)

8. Lưu hình ảnh mới bằng đường dẫn tệp được lưu trữ trong output_path

from PIL import Image

9. Lưu mã dưới dạng background_remover. py

10. Chạy mã bằng cách nhấp vào nút Chạy

11. Chọn hình ảnh mà bạn muốn xóa nền khỏi. Lưu ý rằng hộp thoại có tiêu đề mà chúng tôi đã chỉ định trong mã

12. Điều hướng đến thư mục rembg và đặt tên cho tệp và đặt định dạng tệp thành PNG. Nhấp vào để lưu. Trong ví dụ của chúng tôi, chúng tôi lưu tệp dưới dạng les-no-bg. png

13. Bạn có thể gặp lỗi, nhưng điều này là bình thường. Tải xuống tệp u2net và lưu vào. thư mục u2net trong thư mục người dùng của bạn. Thư mục này được tạo và lưu trữ tự động

Làm cách nào để xóa nền cụ thể khỏi hình ảnh?

Làm cách nào để xóa nền của ảnh trong Pygame?

Làm cách nào để thay đổi nền của hình ảnh thành màu trắng trong Python?