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ů:
- Nahrajte obrázek do služby Google Colab
- Importujte požadované knihovny
- Přečíst vstupní obrázek
- Vypočítejte a vytiskněte velikost vstupního obrázku
- Vytvořte transformaci
- Aplikujte transformaci na zadaný obrázek
- Zobrazte obrázek se změněnou velikostí a jeho velikost
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.sizetisk ( '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.