PyTorch là một framework deep learning nổi tiếng cung cấp nhiều công cụ và mô-đun khác nhau để làm việc với nhiều hình ảnh. Nó cũng cung cấp mô-đun “torchvision.transforms” có một tập hợp các lớp và chức năng để thực hiện các 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. Người dùng có thể sử dụng “ Thay đổi kích thước() ” và chỉ định kích thước, tức là chiều cao và chiều rộng để thay đổi kích thước bất kỳ hình ảnh mong muốn nào. Phương thức này trả về một hình ảnh đã được thay đổi kích thước mới với kích thước đã chỉ định.
Blog này sẽ trình bày phương pháp thay đổi kích thước hình ảnh thành một kích thước cụ thể trong PyTorch.
Làm cách nào để thay đổi kích thước hình ảnh thành kích thước cụ thể trong PyTorch?
Để thay đổi kích thước của hình ảnh và đổi kích thước hình ảnh thành kích thước cụ thể trong PyTorch, hãy làm theo các bước được liệt kê bên dưới:
- Tải hình ảnh lên Google Colab
- Nhập thư viện cần thiết
- Đọc hình ảnh đầu vào
- Tính toán và in kích thước hình ảnh đầu vào
- Tạo một sự biến đổi
- Áp dụng biến đổi trên hình ảnh được chỉ định
- Hiển thị hình ảnh đã thay đổi kích thước và kích thước của nó
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:
Tại đây, chúng tôi đã tải lên hình ảnh sau đây và chúng tôi sẽ thay đổi kích thước của nó thành một kích thước cụ thể:
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ẩunhập khẩu torchvision.transforms BẰNG Biến đổi
từ hình ảnh nhập PIL
nhập matplotlib.pyplot BẰNG làm ơn
Đâ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 ” mở và lưu các định dạng tệp hình ảnh khác nhau.
- “ nhập matplotlib.pyplot dưới dạng plt ” nhập mô-đun “pyplot” được sử dụng để tạo trực quan hóa và biểu đồ:
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 “ galaxy_img.jpg ” và lưu trữ nó trong “ đầu vào_img ' Biến đổi:
input_img = Hình ảnh.open ( 'galaxy_img.jpg' )
Bước 4: Tính và in kích thước ảnh đầu vào
Sau đó, tính toán kích thước của hình ảnh đầu vào, tức là chiều rộng và chiều cao, rồi in nó:
kích cỡ = input_img.sizein ( 'Kích thước của hình ảnh gốc (đầu vào):' , kích cỡ )
Có thể quan sát thấy chiều rộng của hình ảnh đầu vào là “384” và chiều cao của hình ảnh là “576”:
Bước 5: Tạo một sự biến đổi
Tiếp theo, xác định một phép biến đổi để thay đổi kích thước hình ảnh đầu vào. Người dùng cần chỉ định kích thước mới, tức là chiều cao và chiều rộng cho hình ảnh mới. Ở đây, chúng tôi đã chỉ định chiều cao “200” và chiều rộng “400”:
biến đổi = Transform.Resize ( kích cỡ = ( 200 , 400 ) )
Bước 6: Áp dụng Transform trên hình ảnh đầu vào
Bây giờ, áp dụng phép biến đổi ở trên cho hình ảnh đầu vào đã chỉ định để thay đổi kích thước của nó:
new_img = biến đổi ( đầu vào_img )
Bước 7: Hiển thị hình ảnh đã thay đổi kích thước và kích thước của nó
Cuối cùng, xem hình ảnh đã thay đổi kích thước bằng cách hiển thị nó và xác minh kích thước của nó:
in ( 'Kích thước mới của hình ảnh sau khi thay đổi kích thước:' , new_img.size )plt.imshow ( mới_img )
plt.show ( )
Đầu ra bên dưới cho biết hình ảnh đầu vào đã được thay đổi kích thước thành công. Bây giờ, chiều rộng của nó là “400” và chiều cao của nó là “200”:
Tương tự, người dùng có thể chỉ định bất kỳ kích thước nào khác để thay đổi kích thước hình ảnh. Ở đây, chúng tôi sẽ thay đổi kích thước của cùng một hình ảnh đầu vào với kích thước khác, tức là chiều cao “250” và chiều rộng “150”:
biến đổi = Transform.Resize ( kích cỡ = ( 250 , 150 ) )
Điều này sẽ thay đổi kích thước hình ảnh theo kích thước mới:
So sánh
Bạn có thể thấy sự so sánh giữa ảnh gốc và ảnh đã được thay đổi kích thước với các kích thước khác nhau 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 thay đổi kích thước hình ảnh thành kích thước mong muốn trong PyTorch.
Phần kết luận
Để thay đổi kích thước hình ảnh mong muốn thành kích thướ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. Tiếp theo, tính toán và in kích thước của hình ảnh đầu vào. Sau đó, sử dụng “ Thay đổi kích thước() ” để xác định và áp dụng phép biến đổi trên hình ảnh đầu vào mong muốn. Cuối cùng, hiển thị hình ảnh đã được thay đổi kích thước mới và kích thước của nó. Blog này đã trình bày phương pháp thay đổi kích thước hình ảnh mong muốn thành một kích thước cụ thể trong PyTorch.