Làm cách nào để xoay hình ảnh theo góc cụ thể trong PyTorch?

Lam Cach Nao De Xoay Hinh Anh Theo Goc Cu The Trong Pytorch



PyTorch cung cấp mô-đun “torchvision.transforms” có nhiều phép biến đổi cần thiết để thực hiện các tác vụ khác nhau. Mô-đun này cũng có “ Xoay ngẫu nhiên() ” phương pháp áp dụng xoay ngẫu nhiên cho một hình ảnh trong một phạm vi góc xác định theo độ. Biến đổi/phương thức này trả về một hình ảnh được xoay mới.

Bài viết này sẽ minh họa phương pháp xoay một hình ảnh cụ thể theo một góc cụ thể trong PyTorch.

Làm cách nào để xoay hình ảnh mong muốn theo góc cụ thể trong PyTorch?

Để xoay hình ảnh mong muốn theo một góc cụ thể trong PyTorch, hãy xem các bước được liệt kê bên dưới:







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 đây và chúng tôi sẽ xoay nó theo một góc cụ thể:



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:

ngọn đuốc nhập khẩu
nhập khẩu torchvision.transforms BẰNG T
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 T ” 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” được sử dụng để mở và lưu các định dạng tệp hình ả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 “my_img.jpg” và lưu trữ nó trong “đầu vào_img” Biến đổi:

đầu vào_img = Hình ảnh. mở ( 'my_img.jpg' )

Bước 4: Xác định một phép biến đổi

Tiếp theo, xác định một phép biến đổi để xoay hình ảnh đầu vào. Người dùng cần chỉ định phạm vi độ (tối thiểu, tối đa) mong muốn. Ở đây, chúng tôi đã chỉ định (30, 60) độ và hình ảnh mới sẽ được xoay với bất kỳ góc ngẫu nhiên nào được chọn từ phạm vi này:

biến đổi = T. Xoay ngẫu nhiên ( độ = ( 30 , 60 ) )

Bước 5: Áp dụng Transform trên hình ảnh đầu vào

Bây giờ, xoay hình ảnh đầu vào theo các góc được chỉ định bằng cách sử dụng phép biến đổi được xác định ở trên:

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

Bước 6: Hiển thị hình ảnh đã xoay

Cuối cùng, xem hình ảnh được xoay bằng cách hiển thị nó:

đã xoay_img

Kết quả đầu ra trên cho thấy hình ảnh đầu vào đã được xoay thành công theo các góc ngẫu nhiên trong phạm vi quy định là 30 và 60 độ.

Tương tự, người dùng cũng có thể chỉ định các phạm vi độ khác để xoay hình ảnh. Ở đây, chúng tôi sẽ chỉ định phạm vi sau để thấy sự khác biệt:

biến đổi = T. Xoay ngẫu nhiên ( độ = ( 90 , 120 ) )

Điều này sẽ tạo ra một hình ảnh mới sẽ được xoay với bất kỳ góc ngẫu nhiên nào được chọn từ phạm vi được chỉ định là 90 và 120 độ:

So sánh

Có thể thấy sự so sánh giữa ảnh gốc và ảnh được xoay theo các góc khác nhau như 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 xoay hình ảnh theo các góc khác nhau trong PyTorch.

Phần kết luận

Để xoay hình ảnh mong muốn theo một góc cụ thể 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. Sau đó, xác định phép biến đổi bằng cách sử dụng “ Xoay ngẫu nhiên() ” và áp dụng nó cho hình ảnh đầu vào. Cuối cùng hiển thị hình ảnh đã xoay mới. Bài viết này đã minh họa phương pháp xoay hình ảnh mong muốn theo một góc cụ thể trong PyTorch.