Paano I-rotate ang isang Imahe ayon sa Tiyak na Anggulo sa PyTorch?

Paano I Rotate Ang Isang Imahe Ayon Sa Tiyak Na Anggulo Sa Pytorch



Ang PyTorch ay nagbibigay ng module na 'torchvision.transforms' na mayroong maraming mahahalagang pagbabago upang maisagawa ang iba't ibang gawain. Ang modyul na ito ay mayroon ding ' RandomRotation() ” paraan na naglalapat ng random na pag-ikot sa isang imahe sa loob ng isang tinukoy na hanay ng mga anggulo sa antas. Ang pagbabagong ito/paraan ay nagbabalik ng bagong pinaikot na imahe.

Ang artikulong ito ay maglalarawan ng paraan upang paikutin ang isang partikular na imahe sa pamamagitan ng isang tiyak na anggulo sa PyTorch.

Paano I-rotate ang isang Ninanais na Imahe ayon sa Tiyak na Anggulo sa PyTorch?

Upang paikutin ang isang nais na imahe sa pamamagitan ng isang partikular na anggulo sa PyTorch, tingnan ang mga hakbang na nakalista sa ibaba:







Hakbang 1: Mag-upload ng Larawan sa Google Colab

Una, buksan ang Google Colab at mag-click sa mga icon na naka-highlight sa ibaba. Pagkatapos, piliin ang partikular na larawan mula sa computer at i-upload ito:





Sa dakong huli, ang larawan ay ia-upload sa Google Colab:





Dito, na-upload namin ang sumusunod na larawan at iikot namin ito sa isang tiyak na anggulo:



Hakbang 2: Mag-import ng Kinakailangang Library

Pagkatapos nito, i-import ang mga kinakailangang aklatan. Halimbawa, na-import namin ang mga sumusunod na aklatan:

import na tanglaw
import torchvision.transforms bilang T
mula sa PIL import Image

dito:

  • import na tanglaw ” nag-import ng PyTorch library.
  • import torchvision.nagbabago bilang T ” ini-import ang transforms module mula sa torchvision na ginagamit upang iproseso ang data ng imahe bago ito i-feed sa isang neural network.
  • 'mula sa PIL import na Larawan' ay ginagamit para sa pagbubukas at pag-save ng iba't ibang mga format ng file ng imahe:

Hakbang 3: Basahin ang Input na Larawan

Pagkatapos, basahin ang input na imahe mula sa computer. Dito, binabasa natin ang “my_img.jpg” at iniimbak ito sa “input_img” variable:

input_img = Imahe. bukas ( 'my_img.jpg' )

Hakbang 4: Tukuyin ang isang Transform

Susunod, tukuyin ang isang pagbabago upang paikutin ang input na imahe. Kailangang tukuyin ng mga user ang gustong hanay ng (min, max) degrees. Dito, tinukoy namin ang (30, 60) degrees at ang bagong imahe ay iikot sa anumang random na anggulo na pinili mula sa hanay na ito:

ibahin ang anyo = T. RandomRotation ( degrees = ( 30 , 60 ) )

Hakbang 5: Ilapat ang Transform sa Input na Larawan

Ngayon, i-rotate ang input na imahe sa pamamagitan ng tinukoy na mga anggulo gamit ang natukoy na pagbabago sa itaas:

rotated_img = ibahin ang anyo ( input_img )

Hakbang 6: Ipakita ang Pinaikot na Larawan

Panghuli, tingnan ang pinaikot na imahe sa pamamagitan ng pagpapakita nito:

rotated_img

Ang output sa itaas ay nagpapakita na ang input na imahe ay matagumpay na naiikot ng mga random na anggulo sa tinukoy na hanay na 30, at 60 degrees.

Katulad nito, ang mga user ay maaari ring tumukoy ng iba pang mga hanay ng mga degree upang paikutin ang larawan. Dito, tutukuyin namin ang sumusunod na hanay upang makita ang pagkakaiba:

ibahin ang anyo = T. RandomRotation ( degrees = ( 90 , 120 ) )

Ito ay bubuo ng bagong imahe na iikot sa anumang random na anggulo na pinili mula sa tinukoy na hanay na 90 at 120 degrees:

Paghahambing

Ang paghahambing sa pagitan ng orihinal na larawan at mga pinaikot na larawan sa pamamagitan ng iba't ibang mga anggulo ay makikita sa ibaba:

Tandaan : Maa-access mo ang aming Google Colab Notebook dito link .

Mahusay naming naipaliwanag ang paraan upang paikutin ang isang imahe sa pamamagitan ng iba't ibang mga anggulo sa PyTorch.

Konklusyon

Upang i-rotate ang gustong larawan ayon sa isang partikular na anggulo sa PyTorch, una, i-upload ang gustong larawan sa Google Colab. Pagkatapos, i-import ang mga kinakailangang aklatan at basahin ang input na imahe. Pagkatapos nito, tukuyin ang pagbabago gamit ang ' RandomRotation() ” paraan at ilapat ito sa input na imahe. Panghuli, ipakita ang bagong pinaikot na imahe. Ang artikulong ito ay naglalarawan ng paraan upang paikutin ang nais na imahe sa pamamagitan ng isang tiyak na anggulo sa PyTorch.