Kako obrniti sliko za določen kot v PyTorchu?

Kako Obrniti Sliko Za Dolocen Kot V Pytorchu



PyTorch ponuja modul »torchvision.transforms«, ki ima številne bistvene transformacije za izvajanje različnih nalog. Ta modul ima tudi » Naključno Vrtenje() ” metoda, ki uporabi naključno rotacijo slike znotraj določenega obsega kotov v stopinjah. Ta transformacija/metoda vrne novo obrnjeno sliko.

Ta članek bo prikazal metodo zasuka določene slike za določen kot v PyTorchu.

Kako zasukati želeno sliko za določen kot v PyTorchu?

Če želite zasukati želeno sliko za določen kot v PyTorchu, si oglejte spodnje korake:







1. korak: naložite sliko v Google Colab

Najprej odprite Google Colab in kliknite spodaj označene ikone. Nato izberite določeno sliko iz računalnika in jo naložite:





Pozneje bo slika naložena v Google Colab:





Tukaj smo naložili naslednjo sliko in jo bomo zasukali za določen kot:



2. korak: uvozite zahtevano knjižnico

Po tem uvozite potrebne knjižnice. Na primer, uvozili smo naslednje knjižnice:

uvozna svetilka
import torchvision.transforms kot T
iz slike uvoza PIL

Tukaj:

  • uvozna svetilka ” uvozi knjižnico PyTorch.
  • import torchvision.preoblikuje kot T ” uvozi transformatorski modul iz torchvisiona, ki se uporablja za predhodno obdelavo slikovnih podatkov, preden se podajo v nevronsko mrežo.
  • »iz slike uvoza PIL« se uporablja za odpiranje in shranjevanje različnih formatov slikovnih datotek:

3. korak: preberite vhodno sliko

Nato preberite vhodno sliko iz računalnika. Tukaj beremo “moja_img.jpg” in ga shranite v “input_img” spremenljivka:

input_img = Slika. odprto ( 'moja_img.jpg' )

4. korak: Določite transformacijo

Nato definirajte transformacijo za zasuk vhodne slike. Uporabniki morajo določiti želeno območje (najmanj, največ) stopinj. Tu smo določili (30, 60) stopinj in nova slika bo zasukana s katerim koli naključnim kotom, izbranim iz tega obsega:

transformirati = T. Naključno vrtenje ( stopnje = ( 30 , 60 ) )

5. korak: Uporabite transformacijo na vhodni sliki

Zdaj zasukajte vhodno sliko za določene kote z uporabo zgoraj definirane transformacije:

rotated_img = transformirati ( input_img )

6. korak: Prikažite zasukano sliko

Nazadnje si oglejte obrnjeno sliko tako, da jo prikažete:

rotated_img

Zgornji izhod kaže, da je bila vhodna slika uspešno zasukana za naključne kote v določenem obsegu 30 in 60 stopinj.

Podobno lahko uporabniki določijo tudi druge obsege stopinj za vrtenje slike. Tukaj bomo določili naslednji obseg, da vidimo razliko:

transformirati = T. Naključno vrtenje ( stopnje = ( 90 , 120 ) )

To bo ustvarilo novo sliko, ki bo zasukana s katerim koli naključnim kotom, izbranim iz podanega obsega 90 in 120 stopinj:

Primerjava

Primerjavo med originalno sliko in slikami, ki so bile obrnjene pod različnimi koti, si lahko ogledate spodaj:

Opomba : Do našega zvezka Google Colab lahko dostopate tukaj povezava .

Učinkovito smo razložili metodo vrtenja slike pod različnimi koti v PyTorchu.

Zaključek

Če želite zasukati želeno sliko za določen kot v PyTorchu, najprej naložite želeno sliko v Google Colab. Nato uvozite potrebne knjižnice in preberite vhodno sliko. Po tem definirajte transformacijo z uporabo ' Naključno Vrtenje() ” in jo uporabite za vhodno sliko. Na koncu prikažite novo obrnjeno sliko. Ta članek je ponazoril metodo za zasuk želene slike za določen kot v PyTorchu.