Kako spremeniti velikost slike na določeno velikost v PyTorchu?

Kako Spremeniti Velikost Slike Na Doloceno Velikost V Pytorchu



PyTorch je znano ogrodje za globoko učenje, ki ponuja različna orodja in module za delo s številnimi slikami. Zagotavlja tudi modul »torchvision.transforms«, ki ima niz razredov in funkcij za izvajanje različnih transformacij na slikah, kot so obrezovanje, spreminjanje velikosti, vrtenje, obračanje, spreminjanje velikosti in še veliko več. Uporabniki lahko uporabljajo » Spremeni velikost() ” in določite velikost, tj. višino in širino, da spremenite velikost želene slike. Ta metoda vrne novo spremenjeno sliko podane velikosti.

Ta blog bo prikazal način spreminjanja velikosti slike na določeno velikost v PyTorchu.







Kako spremeniti velikost slike na določeno velikost v PyTorchu?

Če želite spremeniti dimenzije slike in jo spremeniti na določeno velikost v PyTorchu, sledite spodnjim korakom:



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 ji bomo spremenili velikost na določeno velikost:

2. korak: uvozite zahtevano knjižnico

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

uvozna svetilka
import torchvision.transforms kot Preobrazba
iz slike uvoza PIL
uvozite matplotlib.pyplot kot plt

Tukaj:

  • uvozna svetilka ” uvozi knjižnico PyTorch.
  • import torchvision.transforms as transforms ” uvozi transformatorski modul iz torchvisiona, ki se uporablja za predhodno obdelavo slikovnih podatkov, preden se podajo v nevronsko mrežo.
  • iz slike uvoza PIL ” odpre in shrani različne formate slikovnih datotek.
  • uvozi matplotlib.pyplot kot plt « uvozi modul »pyplot«, ki se uporablja za ustvarjanje vizualizacij in risb:

3. korak: preberite vhodno sliko

Nato preberite vhodno sliko iz računalnika. Tukaj beremo ' galaxy_img.jpg « in ga shranite v » input_img ” spremenljivka:

input_img = Image.open ( 'galaxy_img.jpg' )

4. korak: Izračunajte in natisnite velikost vhodne slike

Nato izračunajte velikost vhodne slike, tj. širino in višino, ter jo natisnite:

velikost = input_img.size
tiskanje ( 'Velikost izvirne (vhodne) slike:' , velikost )

Opazimo lahko, da je širina vhodne slike '384' in višina slike '576':

5. korak: Ustvarite transformacijo

Nato definirajte transformacijo za spreminjanje velikosti vhodne slike. Uporabniki morajo določiti novo velikost, tj. višino in širino za novo sliko. Tukaj smo določili višino '200' in širino '400':

transform = Transform.Spremeni velikost ( velikost = ( 200 , 400 ) )

6. korak: Uporabite transformacijo na vhodni sliki

Zdaj uporabite zgornjo transformacijo na podani vhodni sliki, da spremenite njeno dimenzijo:

new_img = transformacija ( input_img )

7. korak: Prikažite sliko spremenjene velikosti in njeno velikost

Na koncu si oglejte spremenjeno velikost slike tako, da jo prikažete in preverite njeno velikost:

tiskanje ( 'Nova velikost slike po spremembi velikosti:' , nova_img.velikost )
plt.imshow ( nova_slika )
plt.show ( )

Spodnji rezultat kaže, da je bila velikost vhodne slike uspešno spremenjena. Zdaj je njegova širina '400' in njegova višina '200':

Podobno lahko uporabniki določijo katero koli drugo velikost za spreminjanje velikosti slike. Tukaj bomo spremenili velikost iste vhodne slike z drugo velikostjo, tj. višino »250« in širino »150«:

transform = Transform.Spremeni velikost ( velikost = ( 250 , 150 ) )

S tem boste sliko spremenili v nove dimenzije:

Primerjava

Primerjavo med originalno sliko in slikami spremenjene velikosti z različnimi dimenzijami si lahko ogledate spodaj:

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

Učinkovito smo razložili metodo za spreminjanje velikosti slike na želeno velikost v PyTorchu.

Zaključek

Če želite spremeniti velikost želene slike na določeno velikost v PyTorchu, najprej naložite želeno sliko v Google Colab. Nato uvozite potrebne knjižnice in preberite vhodno sliko. Nato izračunajte in natisnite velikost vhodne slike. Po tem uporabite » Spremeni velikost() ” za definiranje in uporabo transformacije na želeni vhodni sliki. Na koncu prikažite novo spremenjeno sliko in njeno velikost. Ta blog je pokazal metodo za spreminjanje velikosti želene slike na določeno velikost v PyTorchu.