Làm cách nào để cắt bất kỳ hình ảnh nào tại trung tâm của nó ở PyTorch?

Lam Cach Nao De Cat Bat Ky Hinh Anh Nao Tai Trung Tam Cua No O Pytorch



PyTorch cung cấp mô-đun “torchvision.transforms” có một tập hợp các hàm và lớp. Nó thực hiện nhiều phép biến đổi khác nhau trên hình ảnh, chẳng hạn như cắt xén, thay đổi kích thước, xoay, lật, chia tỷ lệ, v.v. Nó cung cấp “ CenterCrop(kích thước) ” phương pháp cắt bất kỳ hình ảnh cụ thể nào ở trung tâm của nó. Người dùng cần chỉ định kích thước cắt mong muốn, tức là chiều cao và chiều rộng của hình ảnh. Phương pháp này tạo ra một hình ảnh được cắt mới có kích thước được chỉ định.

Bài viết này sẽ trình bày phương pháp cắt bất kỳ hình ảnh nào tại trung tâm của nó trong PyTorch.

Làm cách nào để cắt bất kỳ hình ảnh nào tại trung tâm của nó ở PyTorch?

Để cắt bất kỳ hình ảnh nào ở trung tâm của nó trong PyTorch, hãy xem các bước được cung cấp:







  • Tải hình ảnh lên Google Colab
  • Nhập các thư viện cần thiết
  • Đọc hình ảnh đầu vào
  • Xác định một phép biến đổi để cắt hình ảnh đầu vào ở trung tâm của nó
  • Áp dụng biến đổi trên hình ảnh
  • Hiển thị hình ảnh đã cắt

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:





Khi làm như vậy, hình ảnh sẽ được tải lên Google Colab:





Ở đây, chúng tôi đã tải lên hình ảnh sau đây và chúng tôi sẽ cắt nó ở giữa:



Bước 2: Nhập thư viện cần thiết
Tiếp theo, 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:

ngọn đuốc nhập khẩu
nhập khẩu torchvision.transforms BẰNG biến đổi
từ hình ảnh nhập PIL

Đây:

  • ngọn đuốc nhập khẩu ” nhập thư viện PyTorch.
  • nhập torchvision.transforms dưới dạng biến đổi ” nhập mô-đun biến đổi từ torchvision được sử dụng để xử lý trước dữ liệu hình ảnh trước khi đưa nó vào mạng thần kinh.
  • từ hình ảnh nhập PIL ” dùng để mở và lưu các định dạng file ảnh khác nhau:

Bước 3: Đọc hình ảnh đầu vào
Sau đó, đọc hình ảnh đầu vào từ máy tính. Ở đây chúng ta đang đọc “phong cảnh_img.jpg” và lưu trữ nó trong “đầu vào_img” Biến đổi:

input_img = Hình ảnh.open ( 'phong cảnh_img.jpg' )

Bước 4: Xác định một phép biến đổi
Sau đó, xác định một phép biến đổi để cắt hình ảnh đầu vào ở trên ở giữa. Người dùng có thể cung cấp một giá trị cho phần cắt hình vuông hoặc hai giá trị cho phần cắt hình chữ nhật. Ở đây, chúng tôi đã chỉ định một giá trị, tức là “230” cho hình vuông:

biến đổi = biến đổi.CenterCrop ( ( 230 ) )

Bước 5: Áp dụng Transform trên hình ảnh
Bây giờ, áp dụng phép biến đổi ở trên cho hình ảnh đầu vào mong muốn để cắt nó ở giữa:

cropped_img = biến đổi ( đầu vào_img )

Bước 6: Hiển thị hình ảnh đã cắt
Cuối cùng, xem hình ảnh đã cắt bằng cách hiển thị nó:

đã cắt_img

Kết quả đầu ra trên cho thấy hình ảnh đầu vào đã được cắt thành công ở giữa thành hình vuông.

Tương tự, người dùng cũng có thể chỉ định hai giá trị là chiều cao và chiều rộng trong “CenterCrop()” phương pháp cắt hình ảnh ở trung tâm của nó thành hình chữ nhật. Ở đây, chúng tôi sẽ cung cấp chiều cao và chiều rộng như sau:

biến đổi = biến đổi.CenterCrop ( ( 150 , 280 ) )

Điều này đã cắt hình ảnh thành hình chữ nhật:

So sánh
Bạn có thể thấy sự so sánh giữa ảnh gốc và ảnh đã crop 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 một cách hiệu quả phương pháp cắt xén hình ảnh ở trung tâm PyTorch.

Phần kết luận

Để cắt một hình ảnh cụ thể ở trung tâm của nó 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 “CenterCrop()” phương pháp cắt hình ảnh đầu vào mong muốn ở trung tâm của nó. Cuối cùng, xem hình ảnh đã cắt bằng cách hiển thị nó. Bài viết này trình bày phương pháp cắt bất kỳ hình ảnh nào tại trung tâm của nó trong PyTorch.