Jak změnit velikost obrázku na konkrétní velikost v PyTorch?

Jak Zmenit Velikost Obrazku Na Konkretni Velikost V Pytorch



PyTorch je známý framework pro hluboké učení, který nabízí různé nástroje a moduly pro práci s četnými obrázky. Poskytuje také modul „torchvision.transforms“, který má sadu tříd a funkcí pro provádění různých transformací obrázků, jako je oříznutí, změna velikosti, otočení, převrácení, změna měřítka a mnoho dalších. Uživatelé mohou použít „ Změnit velikost () ” a zadejte velikost, tj. výšku a šířku, abyste mohli změnit velikost libovolného obrázku. Tato metoda vrátí nový obrázek se změněnou velikostí zadané velikosti.

Tento blog bude demonstrovat metodu změny velikosti obrázku na konkrétní velikost v PyTorch.







Jak změnit velikost obrázku na konkrétní velikost v PyTorch?

Chcete-li změnit rozměry obrázku a změnit jeho velikost na konkrétní velikost v PyTorch, postupujte podle níže uvedených kroků:



Krok 1: Nahrajte obrázek do služby Google Colab



Nejprve otevřete Google Colab a klikněte na níže zvýrazněné ikony. Poté vyberte konkrétní obrázek z počítače a nahrajte jej:





Následně bude obrázek nahrán do služby Google Colab:



Zde jsme nahráli následující obrázek a změníme jeho velikost na konkrétní velikost:

Krok 2: Importujte požadovanou knihovnu

Dále importujte potřebné knihovny. Importovali jsme například následující knihovny:

dovozní pochodeň
importovat torchvision.transformuje tak jako Přeměnit
z importu PIL Obrázek
import matplotlib.pyplot tak jako plt

Tady:

  • dovozní pochodeň ” importuje knihovnu PyTorch.
  • import torchvision.transforms as transforms ” importuje transformační modul z torchvision, který se používá k předběžnému zpracování obrazových dat před jejich dodáním do neuronové sítě.
  • z importu PIL Obrázek ” otevře a uloží různé formáty obrazových souborů.
  • import matplotlib.pyplot jako plt ” importuje modul „pyplot“, který se používá k vytváření vizualizací a grafů:

Krok 3: Přečtěte si vstupní obrázek

Poté načtěte vstupní obrázek z počítače. Zde čteme „ galaxy_img.jpg “ a jeho uložením do „ input_img 'proměnná:

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

Krok 4: Vypočítejte a vytiskněte vstupní velikost obrázku

Poté spočítejte velikost vstupního obrázku, tj. šířku a výšku, a vytiskněte jej:

velikost = input_img.size
tisk ( 'Velikost původního (vstupního) obrázku:' , velikost )

Lze pozorovat, že šířka vstupního obrázku je „384“ a výška obrázku je „576“:

Krok 5: Vytvořte transformaci

Dále definujte transformaci pro změnu velikosti vstupního obrázku. Uživatelé musí zadat novou velikost, tj. výšku a šířku nového obrázku. Zde jsme zadali výšku „200“ a šířku „400“:

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

Krok 6: Aplikujte transformaci na vstupní obrázek

Nyní aplikujte výše uvedenou transformaci na zadaný vstupní obrázek, abyste změnili jeho rozměr:

new_img = transformace ( input_img )

Krok 7: Zobrazte obrázek se změněnou velikostí a jeho velikost

Nakonec si zobrazte obrázek se změněnou velikostí a ověřte jeho velikost:

tisk ( 'Nová velikost obrázku po změně velikosti:' , new_img.size )
plt.imshow ( new_img )
plt.show ( )

Níže uvedený výstup znamená, že velikost vstupního obrázku byla úspěšně změněna. Nyní je jeho šířka „400“ a jeho výška je „200“:

Podobně mohou uživatelé určit jakoukoli jinou velikost pro změnu velikosti obrázku. Zde změníme velikost stejného vstupního obrázku s jinou velikostí, tj. na výšku „250“ a šířku „150“:

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

Tím se změní velikost obrázku na nové rozměry:

Srovnání

Porovnání mezi původním obrázkem a obrázky se změněnou velikostí s různými rozměry lze vidět níže:

Poznámka : Zde můžete přistupovat k našemu Zápisníku Google Colab odkaz .

Účinně jsme vysvětlili metodu změny velikosti obrázku na požadovanou velikost v PyTorch.

Závěr

Chcete-li v PyTorch změnit velikost požadovaného obrázku na konkrétní velikost, nejprve požadovaný obrázek nahrajte do služby Google Colab. Poté importujte potřebné knihovny a přečtěte si vstupní obrázek. Dále vypočítejte a vytiskněte velikost vstupního obrázku. Poté použijte „ Změnit velikost () ” metodu k definování a aplikaci transformace na požadovaný vstupní obraz. Nakonec zobrazte nový obrázek se změněnou velikostí a jeho velikost. Tento blog demonstroval metodu změny velikosti požadovaného obrázku na konkrétní velikost v PyTorch.