Kako rotirati sliku za određeni kut u PyTorchu?

Kako Rotirati Sliku Za Odredeni Kut U Pytorchu



PyTorch pruža modul 'torchvision.transforms' koji ima mnogo bitnih transformacija za obavljanje različitih zadataka. Ovaj modul također ima ' Slučajna rotacija() ” metoda koja primjenjuje nasumičnu rotaciju na sliku unutar određenog raspona kutova u stupnjevima. Ova transformacija/metoda vraća novu zakrenutu sliku.

Ovaj će članak ilustrirati metodu rotiranja određene slike za određeni kut u PyTorchu.

Kako rotirati željenu sliku za određeni kut u PyTorchu?

Za rotiranje željene slike za određeni kut u PyTorchu, provjerite dolje navedene korake:







Korak 1: Prenesite sliku na Google Colab

Prvo otvorite Google Colab i kliknite dolje označene ikone. Zatim odaberite određenu sliku s računala i prenesite je:





Nakon toga, slika će biti prenesena na Google Colab:





Ovdje smo učitali sljedeću sliku i rotirati ćemo je za određeni kut:



Korak 2: Uvezite potrebnu biblioteku

Nakon toga uvezite potrebne biblioteke. Na primjer, uvezli smo sljedeće biblioteke:

uvozna baklja
import torchvision.preobražava kao T
iz PIL import slike

Ovdje:

  • uvozna baklja ” uvozi biblioteku PyTorch.
  • import torchvision.pretvara se kao T ” uvozi transformacijski modul iz torchvisiona koji se koristi za prethodnu obradu slikovnih podataka prije nego što se unesu u neuronsku mrežu.
  • “iz PIL uvozne slike” koristi se za otvaranje i spremanje različitih formata slikovnih datoteka:

Korak 3: Pročitajte ulaznu sliku

Zatim pročitajte ulaznu sliku s računala. Evo, mi čitamo “moja_img.jpg” i pohranjujući ga u “input_img” varijabla:

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

Korak 4: Definirajte transformaciju

Zatim definirajte transformaciju za rotiranje ulazne slike. Korisnici moraju navesti željeni raspon (min, maks.) stupnjeva. Ovdje smo odredili (30, 60) stupnjeva i nova slika će se rotirati pod bilo kojim slučajnim kutom odabranim iz ovog raspona:

transformirati = T. Slučajna rotacija ( stupnjeva = ( 30 , 60 ) )

Korak 5: Primijenite transformaciju na ulaznu sliku

Sada rotirajte ulaznu sliku za određene kutove koristeći gore definiranu transformaciju:

rotated_img = transformirati ( input_img )

Korak 6: Prikažite rotiranu sliku

Na kraju, pogledajte zakrenutu sliku tako da je prikažete:

rotated_img

Gornji izlaz pokazuje da je ulazna slika uspješno rotirana za nasumične kutove u navedenom rasponu od 30 i 60 stupnjeva.

Slično, korisnici također mogu odrediti druge raspone stupnjeva za rotiranje slike. Ovdje ćemo navesti sljedeći raspon kako bismo vidjeli razliku:

transformirati = T. Slučajna rotacija ( stupnjeva = ( 90 , 120 ) )

Ovo će generirati novu sliku koja će se rotirati pod bilo kojim slučajnim kutom odabranim iz navedenog raspona od 90 i 120 stupnjeva:

Usporedba

Usporedba između izvorne slike i slika zakrenutih pod različitim kutovima može se vidjeti u nastavku:

Bilješka : Našoj Google Colab bilježnici možete pristupiti ovdje veza .

Učinkovito smo objasnili metodu rotiranja slike pod različitim kutovima u PyTorchu.

Zaključak

Za rotiranje željene slike za određeni kut u PyTorchu, prvo prenesite željenu sliku na Google Colab. Zatim uvezite potrebne biblioteke i pročitajte ulaznu sliku. Nakon toga definirajte transformaciju pomoću ' Slučajna rotacija() ” i primijenite je na ulaznu sliku. Na kraju, prikažite novu rotiranu sliku. Ovaj je članak ilustrirao metodu rotiranja željene slike za određeni kut u PyTorchu.