Bài viết này sẽ trình bày phương pháp thay đổi màu sắc của hình ảnh trong PyTorch.
Làm cách nào để điều chỉnh/thay đổi màu sắc của hình ảnh trong PyTorch?
Để điều chỉnh màu sắc của hình ảnh trong PyTorch, hãy xem các bước sau:
- Tải hình ảnh mong muốn lên Google Colab
- Nhập các thư viện cần thiết
- Đọc hình ảnh đầu vào
- Thay đổi màu sắc của hình ảnh đầu vào
- Hiển thị hình ảnh được điều chỉnh màu sắc
Bước 1: Tải hình ảnh lên Google Colab
Đầu tiên, hãy mở Google Colab và nhấp vào các biểu tượng được đánh dấu bên dưới. Sau đó, chọn hình ảnh cụ thể từ máy tính và tải lên:
Sau đó, hình ảnh sẽ được tải lên Google Colab:
Ở đây, chúng tôi đã tải lên hình ảnh sau và chúng tôi sẽ điều chỉnh màu sắc của hình ảnh này:
Bước 2: Nhập thư viện cần thiết
Sau đó, nhập các thư viện cần thiết. Ví dụ: chúng tôi đã nhập các thư viện sau:
nhập khẩu ngọn đuốctừ PIL nhập khẩu Hình ảnh
nhập khẩu torchvision. biến đổi . chức năng BẰNG F
Đây:
- “ ngọn đuốc nhập khẩu ” nhập thư viện PyTorch.
- “ từ hình ảnh nhập PIL ” được sử dụng để mở và lưu các định dạng tệp hình ảnh khác nhau.
- “ nhập torchvision.transforms.function dưới dạng F ” nhập mô-đun chức năng từ “torchvision.transforms” cung cấp các phép biến đổi:
Bước 3: Đọc hình ảnh đầu vào
Tiếp theo, đọc hình ảnh đầu vào từ máy tính. Ở đây chúng ta đang đọc “ hoa_img.jpg ” và lưu trữ nó trong “ đầu vào_img ' Biến đổi:
đầu vào_img = Hình ảnh. mở ( 'hoa_img.jpg' )Bước 4: Thay đổi màu sắc của hình ảnh đầu vào
Bây giờ, hãy sửa đổi màu sắc của hình ảnh đầu vào bằng hệ số màu cụ thể bằng cách sử dụng nút “ điều chỉnh_hue() ' phương pháp. Ở đây, chúng tôi đang điều chỉnh màu sắc bằng hệ số “ 0,2 ”:
mới_img = F. điều chỉnh_hue ( đầu vào_img , 0,2 )Bước 5: Hiển thị hình ảnh đã điều chỉnh Hue
Cuối cùng, xem hình ảnh đã điều chỉnh màu sắc bằng cách hiển thị nó:
mới_img
Kết quả đầu ra ở trên cho thấy màu sắc của hình ảnh đầu vào đã được điều chỉnh thành công với hệ số màu sắc được chỉ định, tức là “0,2”.
Tương tự, người dùng có thể chỉ định bất kỳ hệ số màu nào khác để điều chỉnh màu sắc của hình ảnh. Bây giờ, chúng ta sẽ điều chỉnh hình ảnh tương tự bằng nút “ -0,3 ” Yếu tố màu sắc để thấy sự khác biệt:
mới_img = F. điều chỉnh_hue ( đầu vào_img , - 0,3 )Điều này sẽ thay đổi màu sắc của hình ảnh:
Ghi chú: Người dùng có thể cung cấp giá trị hệ số màu trong phạm vi [−0,5, 0,5]. Các giá trị này mang lại cho hình ảnh các màu bổ sung trong khi “0” mang lại hình ảnh gốc.
So sánh
Có thể thấy sự so sánh giữa ảnh gốc và ảnh đã điều chỉnh màu sắc dưới đây:
Ghi chú : Bạn có thể truy cập Sổ tay Google Colab của chúng tôi tại đây liên kết .
Chúng tôi đã giải thích phương pháp hiệu quả để điều chỉnh màu sắc của hình ảnh trong PyTorch.
Phần kết luận
Để điều chỉnh/thay đổi màu sắc của hình ảnh trong PyTorch, trước tiên, hãy tải hình ảnh mong muốn lên Google Colab. Sau đó, nhập các thư viện cần thiết và đọc hình ảnh đầu vào. Tiếp theo, sử dụng “ điều chỉnh_hue() ” để thay đổi màu sắc của hình ảnh đầu vào với các hệ số màu mong muốn. Cuối cùng, xem hình ảnh đã điều chỉnh màu sắc bằng cách hiển thị nó. Bài viết này đã trình bày phương pháp điều chỉnh/thay đổi màu sắc của hình ảnh trong PyTorch.