Kaip „PyTorch“ pakeisti vaizdo dydį iki konkretaus dydžio?

Kaip Pytorch Pakeisti Vaizdo Dydi Iki Konkretaus Dydzio



„PyTorch“ yra gerai žinoma gilaus mokymosi sistema, siūlanti įvairius įrankius ir modulius darbui su daugybe vaizdų. Jame taip pat yra modulis „torchvision.transforms“, kuriame yra klasių ir funkcijų rinkinys, skirtas įvairioms vaizdų transformacijoms atlikti, pavyzdžiui, apkarpyti, keisti dydį, pasukti, apversti, keisti mastelį ir daug daugiau. Vartotojai gali naudoti „ Keisti dydį () “ metodą ir nurodykite dydį, t. y. aukštį ir plotį, kad pakeistumėte norimo vaizdo dydį. Šis metodas grąžina naują pakeisto dydžio nurodyto dydžio vaizdą.

Šiame tinklaraštyje bus parodytas vaizdo dydžio keitimo iki konkretaus dydžio metodas „PyTorch“.







Kaip „PyTorch“ pakeisti vaizdo dydį iki konkretaus dydžio?

Norėdami pakeisti vaizdo matmenis ir pakeisti jo dydį iki konkretaus dydžio „PyTorch“, atlikite toliau nurodytus veiksmus.



1 veiksmas: įkelkite vaizdą į „Google Colab“.



Pirmiausia atidarykite „Google Colab“ ir spustelėkite toliau paryškintas piktogramas. Tada pasirinkite konkretų vaizdą iš kompiuterio ir įkelkite jį:





Vėliau vaizdas bus įkeltas į „Google Colab“:



Čia mes įkėlėme šį vaizdą ir pakeisime jo dydį iki konkretaus dydžio:

2 veiksmas: importuokite reikiamą biblioteką

Tada importuokite reikiamas bibliotekas. Pavyzdžiui, importavome šias bibliotekas:

importo žibintuvėlis
importuoti torchvision.transformuoja kaip Transformuoti
iš PIL importo paveikslėlio
importuoti matplotlib.pyplot kaip plt

Čia:

  • importo žibintuvėlis “ importuoja PyTorch biblioteką.
  • importuoti torchvision.transforms as transforms “ importuoja transformavimo modulį iš „torchvision“, kuris naudojamas vaizdo duomenims iš anksto apdoroti prieš perduodant juos į neuroninį tinklą.
  • iš PIL importo paveikslėlio “ atidaro ir išsaugo skirtingus vaizdo failų formatus.
  • importuoti matplotlib.pyplot kaip plt “ importuoja „pyplot“ modulį, kuris naudojamas vizualizacijoms ir brėžiniams kurti:

3 veiksmas: perskaitykite įvesties vaizdą

Po to perskaitykite įvesties vaizdą iš kompiuterio. Čia mes skaitome „ galaxy_img.jpg “ ir saugoti jį „ input_img “ kintamasis:

input_img = Vaizdas.atviras ( „galaxy_img.jpg“ )

4 veiksmas: apskaičiuokite ir spausdinkite įvesties vaizdo dydį

Tada apskaičiuokite įvesties vaizdo dydį, ty plotį ir aukštį, ir atspausdinkite:

dydis = input_img.size
spausdinti ( 'Pradinio (įvesties) vaizdo dydis:' , dydis )

Galima pastebėti, kad įvesties vaizdo plotis yra „384“, o vaizdo aukštis yra „576“:

5 veiksmas: sukurkite transformaciją

Tada nustatykite transformaciją, kad pakeistumėte įvesties vaizdo dydį. Vartotojai turi nurodyti naują dydį, ty naujo vaizdo aukštį ir plotį. Čia mes nurodėme aukštį „200“ ir plotį „400“:

transformuoti = Transform.Resize ( dydis = ( 200 , 400 ) )

6 veiksmas: pritaikykite transformaciją įvesties vaizde

Dabar pritaikykite aukščiau pateiktą transformaciją nurodytam įvesties vaizdui, kad pakeistumėte jo matmenis:

new_img = transformuoti ( input_img )

7 veiksmas: parodykite pakeisto dydžio vaizdą ir jo dydį

Galiausiai peržiūrėkite pakeisto dydžio vaizdą rodydami jį ir patikrinkite jo dydį:

spausdinti ( 'Naujas vaizdo dydis pakeitus dydį:' , new_img.size )
plt.imshow ( naujas_img )
plt.show ( )

Žemiau pateikta išvestis rodo, kad įvesties vaizdo dydis buvo sėkmingai pakeistas. Dabar jo plotis yra '400', o aukštis - '200':

Panašiai vartotojai gali nurodyti bet kokį kitą dydį, kad pakeistų vaizdo dydį. Čia pakeisime to paties įvesties vaizdo dydį, naudodami kitą dydį, ty aukštį „250“ ir plotį „150“:

transformuoti = Transform.Resize ( dydis = ( 250 , 150 ) )

Tai pakeis vaizdo dydį į naujus matmenis:

Palyginimas

Žemiau galite pamatyti originalaus vaizdo ir pakeistų skirtingų matmenų vaizdų palyginimą:

Pastaba : „Google Colab“ bloknotą galite pasiekti šiuo adresu nuoroda .

Mes efektyviai paaiškinome, kaip „PyTorch“ pakeisti vaizdo dydį iki norimo dydžio.

Išvada

Norėdami pakeisti norimo vaizdo dydį iki konkretaus dydžio „PyTorch“, pirmiausia įkelkite norimą vaizdą į „Google Colab“. Tada importuokite reikiamas bibliotekas ir perskaitykite įvesties vaizdą. Tada apskaičiuokite ir atspausdinkite įvesties vaizdo dydį. Po to naudokite „ Keisti dydį () “ metodą, norėdami apibrėžti ir pritaikyti transformaciją norimam įvesties vaizdui. Galiausiai parodykite naują pakeisto dydžio vaizdą ir jo dydį. Šis tinklaraštis demonstravo būdą, kaip pakeisti norimo vaizdo dydį iki konkretaus dydžio „PyTorch“.