Kā PyTorch pagriezt attēlu pēc noteikta leņķa?

Ka Pytorch Pagriezt Attelu Pec Noteikta Lenka



PyTorch nodrošina moduli 'torchvision.transforms', kurā ir daudz būtisku pārveidojumu dažādu uzdevumu veikšanai. Šim modulim ir arī ' RandomRotation() ” metode, kas attēlam piemēro nejaušu pagriešanu noteiktā leņķu diapazonā. Šī transformācija/metode atgriež jaunu pagrieztu attēlu.

Šajā rakstā tiks parādīta metode, kā PyTorch pagriezt noteiktu attēlu noteiktā leņķī.

Kā PyTorch pagriezt vēlamo attēlu pēc noteikta leņķa?

Lai PyTorch pagrieztu vēlamo attēlu noteiktā leņķī, veiciet tālāk norādītās darbības.







1. darbība. Augšupielādējiet attēlu pakalpojumā Google Colab

Vispirms atveriet Google Colab un noklikšķiniet uz tālāk izceltajām ikonām. Pēc tam datorā izvēlieties konkrēto attēlu un augšupielādējiet to:





Pēc tam attēls tiks augšupielādēts pakalpojumā Google Colab:





Šeit mēs esam augšupielādējuši šādu attēlu un pagriezīsim to noteiktā leņķī:



2. darbība. Importējiet nepieciešamo bibliotēku

Pēc tam importējiet nepieciešamās bibliotēkas. Piemēram, mēs esam importējuši šādas bibliotēkas:

importa lāpa
importēt torchvision.transforms T
no PIL importa attēla

Šeit:

  • ' importa lāpa ” importē PyTorch bibliotēku.
  • ' importēt torchvision.pārveidojas kā T ” importē pārveidošanas moduli no torchvision, ko izmanto attēla datu pirmapstrādei pirms to ievadīšanas neironu tīklā.
  • “no PIL importa attēla” tiek izmantots dažādu attēlu failu formātu atvēršanai un saglabāšanai:

3. darbība. Izlasiet ievades attēlu

Pēc tam izlasiet ievades attēlu no datora. Šeit mēs lasām “mans_img.jpg” un uzglabājot to “input_img” mainīgais:

ievades_img = Attēls. atvērts ( 'my_img.jpg' )

4. darbība: definējiet transformāciju

Pēc tam definējiet transformāciju, lai pagrieztu ievades attēlu. Lietotājiem ir jānorāda vēlamais (min, max) grādu diapazons. Šeit mēs esam norādījuši (30, 60) grādus, un jaunais attēls tiks pagriezts ar jebkuru nejauši izvēlētu leņķi no šī diapazona:

pārveidot = T. RandomRotation ( grādiem = ( 30 , 60 ) )

5. darbība. Lietojiet transformāciju ievades attēlā

Tagad pagrieziet ievades attēlu par norādītajiem leņķiem, izmantojot iepriekš definēto transformāciju:

rotated_img = pārveidot ( ievades_img )

6. darbība. Parādiet pagrieztu attēlu

Visbeidzot, apskatiet pagriezto attēlu, parādot to:

rotated_img

Iepriekš redzamā izvade parāda, ka ievades attēls ir veiksmīgi pagriezts nejaušos leņķos norādītajā diapazonā no 30 līdz 60 grādiem.

Tāpat lietotāji attēla pagriešanai var norādīt arī citus grādu diapazonus. Šeit mēs norādīsim šādu diapazonu, lai redzētu atšķirību:

pārveidot = T. RandomRotation ( grādiem = ( 90 , 120 ) )

Tādējādi tiks izveidots jauns attēls, kas tiks pagriezts ar jebkuru nejaušu leņķi, kas atlasīts no norādītā diapazona no 90 līdz 120 grādiem:

Salīdzinājums

Salīdzinājumu starp oriģinālo attēlu un pagrieztiem attēliem dažādos leņķos var redzēt zemāk:

Piezīme : varat piekļūt mūsu Google Colab piezīmju grāmatiņai šeit saite .

Mēs esam efektīvi izskaidrojuši metodi attēla pagriešanai dažādos leņķos programmā PyTorch.

Secinājums

Lai pakalpojumā PyTorch pagrieztu vajadzīgo attēlu noteiktā leņķī, vispirms augšupielādējiet vajadzīgo attēlu pakalpojumā Google Colab. Pēc tam importējiet vajadzīgās bibliotēkas un izlasiet ievades attēlu. Pēc tam definējiet transformāciju, izmantojot ' RandomRotation() ” metodi un lietojiet to ievades attēlam. Visbeidzot parādiet jauno pagriezto attēlu. Šajā rakstā ir ilustrēta metode, kā PyTorch pagriezt vēlamo attēlu noteiktā leņķī.