PyTorch je známy rámec pre hlboké učenie, ktorý ponúka rôzne nástroje a moduly na prácu s množstvom obrázkov. Poskytuje tiež modul „torchvision.transforms“, ktorý má sadu tried a funkcií na vykonávanie rôznych transformácií obrázkov, ako je orezanie, zmena veľkosti, otočenie, prevrátenie, zmena mierky a mnoho ďalších. Používatelia môžu použiť „ Zmeniť veľkosť () “ a zadajte veľkosť, t. j. výšku a šírku, aby ste zmenili veľkosť ľubovoľného obrázka. Táto metóda vráti nový obrázok so zmenenou veľkosťou zadanej veľkosti.
Tento blog bude demonštrovať metódu zmeny veľkosti obrázka na konkrétnu veľkosť v PyTorch.
Ako zmeniť veľkosť obrázka na konkrétnu veľkosť v PyTorch?
Ak chcete zmeniť rozmery obrázka a zmeniť jeho veľkosť na konkrétnu veľkosť v PyTorch, postupujte podľa krokov uvedených nižšie:
- Nahrajte obrázok do služby Google Colab
- Importujte požadované knižnice
- Prečítajte si vstupný obrázok
- Vypočítajte a vytlačte veľkosť vstupného obrázka
- Vytvorte transformáciu
- Použiť transformáciu na zadaný obrázok
- Zobrazte obrázok so zmenenou veľkosťou a jeho veľkosť
Krok 1: Nahrajte obrázok do služby Google Colab
Najprv otvorte Google Colab a kliknite na nižšie zvýraznené ikony. Potom vyberte konkrétny obrázok z počítača a nahrajte ho:
Následne sa obrázok nahrá do služby Google Colab:
Tu sme odovzdali nasledujúci obrázok a jeho veľkosť zmeníme na konkrétnu veľkosť:
Krok 2: Importujte požadovanú knižnicu
Ďalej importujte potrebné knižnice. Napríklad sme importovali nasledujúce knižnice:
dovozová baterkaimportovať torchvision.transformuje ako Transformovať
z importu PIL Obrázok
import matplotlib.pyplot ako plt
Tu:
- “ dovozová baterka importuje knižnicu PyTorch.
- “ import torchvision.transformuje ako transformuje ” importuje transformačný modul z torchvision, ktorý sa používa na predbežné spracovanie obrazových údajov pred ich privedením do neurónovej siete.
- “ z importu PIL Obrázok “ otvorí a uloží rôzne formáty obrázkových súborov.
- “ import matplotlib.pyplot as plt ” importuje modul “pyplot”, ktorý sa používa na vytváranie vizualizácií a grafov:
Krok 3: Prečítajte si vstupný obrázok
Potom si prečítajte vstupný obrázok z počítača. Tu čítame „ galaxy_img.jpg “ a jeho uložením do „ input_img 'premenná:
input_img = Image.open ( 'galaxy_img.jpg' )
Krok 4: Vypočítajte a vytlačte vstupnú veľkosť obrázka
Potom vypočítajte veľkosť vstupného obrázka, t. j. šírku a výšku, a vytlačte ho:
veľkosť = input_img.sizevytlačiť ( 'Veľkosť pôvodného (vstupného) obrázku:' , veľkosť )
Je možné pozorovať, že šírka vstupného obrázka je „384“ a výška obrázka je „576“:
Krok 5: Vytvorte transformáciu
Ďalej definujte transformáciu na zmenu veľkosti vstupného obrázka. Používatelia musia zadať novú veľkosť, tj výšku a šírku nového obrázka. Tu sme zadali výšku „200“ a šírku „400“:
transform = Transform.Resize ( veľkosť = ( 200 , 400 ) )
Krok 6: Použite transformáciu na vstupný obrázok
Teraz aplikujte vyššie uvedenú transformáciu na zadaný vstupný obrázok, aby ste zmenili jeho rozmer:
new_img = transformácia ( input_img )
Krok 7: Zobrazte obrázok so zmenenou veľkosťou a jeho veľkosť
Nakoniec si zobrazte obrázok so zmenenou veľkosťou a overte jeho veľkosť:
vytlačiť ( 'Nová veľkosť obrázka po zmene veľkosti:' , new_img.size )plt.imshow ( new_img )
plt.show ( )
Nižšie uvedený výstup znamená, že veľkosť vstupného obrázka bola úspešne zmenená. Teraz je jeho šírka „400“ a jeho výška je „200“:
Podobne môžu používatelia zadať akúkoľvek inú veľkosť na zmenu veľkosti obrázka. Tu zmeníme veľkosť rovnakého vstupného obrázka s inou veľkosťou, t. j. na výšku „250“ a šírku „150“:
transform = Transform.Resize ( veľkosť = ( 250 , 150 ) )
Toto zmení veľkosť obrázka na nové rozmery:
Porovnanie
Porovnanie medzi pôvodným obrázkom a obrázkami so zmenenou veľkosťou s rôznymi rozmermi si môžete pozrieť nižšie:
Poznámka : K nášmu Zápisníku Google Colab môžete pristupovať tu odkaz .
Účinne sme vysvetlili metódu zmeny veľkosti obrázka na požadovanú veľkosť v PyTorch.
Záver
Ak chcete zmeniť veľkosť požadovaného obrázka na konkrétnu veľkosť v PyTorch, najprv nahrajte požadovaný obrázok do služby Google Colab. Potom importujte potrebné knižnice a prečítajte si vstupný obrázok. Ďalej vypočítajte a vytlačte veľkosť vstupného obrázka. Potom použite „ Zmeniť veľkosť () ” na definovanie a aplikovanie transformácie na požadovaný vstupný obrázok. Nakoniec zobrazte nový obrázok so zmenenou veľkosťou a jeho veľkosť. Tento blog demonštroval metódu na zmenu veľkosti požadovaného obrázka na konkrétnu veľkosť v PyTorch.