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:
- Naložite sliko v Google Colab
- Uvoz zahtevanih knjižnic
- Preberite vhodno sliko
- Izračunajte in natisnite velikost vhodne slike
- Ustvari preobrazbo
- Uporabite transformacijo na določeni sliki
- Prikaži spremenjeno velikost slike in njeno velikost
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 svetilkaimport 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.sizetiskanje ( '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.