PyTorch é uma conhecida estrutura de aprendizado profundo que oferece várias ferramentas e módulos para trabalhar com inúmeras imagens. Ele também fornece um módulo “torchvision.transforms” que possui um conjunto de classes e funções para realizar diversas transformações em imagens, como cortar, redimensionar, girar, inverter, dimensionar e muito mais. Os usuários podem usar o “ Redimensionar() ”E especifique o tamanho, ou seja, altura e largura para redimensionar qualquer imagem desejada. Este método retorna uma nova imagem redimensionada com o tamanho especificado.
Este blog demonstrará o método de redimensionar uma imagem para um tamanho específico no PyTorch.
Como redimensionar uma imagem para um tamanho específico no PyTorch?
Para alterar as dimensões de uma imagem e redimensioná-la para um tamanho específico no PyTorch, siga as etapas listadas abaixo:
- Faça upload de uma imagem para o Google Colab
- Importe as bibliotecas necessárias
- Ler imagem de entrada
- Calcular e imprimir o tamanho da imagem de entrada
- Crie uma transformação
- Aplicar transformação na imagem especificada
- Exibir a imagem redimensionada e seu tamanho
Etapa 1: faça upload de uma imagem para o Google Colab
Primeiro, abra o Google Colab e clique nos ícones destacados abaixo. Em seguida, escolha a imagem específica do computador e carregue-a:
Posteriormente, a imagem será carregada no Google Colab:
Aqui, carregamos a seguinte imagem e iremos redimensioná-la para um tamanho específico:
Etapa 2: importar a biblioteca necessária
A seguir, importe as bibliotecas necessárias. Por exemplo, importamos as seguintes bibliotecas:
importar tochaimportar torchvision.transforms como Transformar
da imagem de importação PIL
importar matplotlib.pyplot como plt
Aqui:
- “ importar tocha ”Importa a biblioteca PyTorch.
- “ importar torchvision.transforms como transformações ”importa o módulo de transformação do torchvision que é usado para pré-processar dados de imagem antes de alimentá-los em uma rede neural.
- “ da imagem de importação PIL ”Abre e salva diferentes formatos de arquivo de imagem.
- “ importar matplotlib.pyplot como plt ”importa o módulo “pyplot” que é usado para criar visualizações e gráficos:
Etapa 3: leia a imagem de entrada
Depois disso, leia a imagem de entrada do computador. Aqui, estamos lendo o “ galáxia_img.jpg ”E armazená-lo no“ entrada_img ' variável:
input_img = Imagem.abrir ( 'galaxy_img.jpg' )
Etapa 4: calcular e imprimir o tamanho da imagem de entrada
Em seguida, calcule o tamanho da imagem de entrada, ou seja, largura e altura, e imprima-a:
tamanho = input_img.tamanhoimprimir ( 'Tamanho da imagem original (entrada):' , tamanho )
Pode-se observar que a largura da imagem de entrada é “384” e a altura da imagem é “576”:
Etapa 5: crie uma transformação
A seguir, defina uma transformação para redimensionar a imagem de entrada. Os usuários precisam especificar o novo tamanho, ou seja, altura e largura da nova imagem. Aqui, especificamos a altura “200” e a largura “400”:
transformar = Transformar.Redimensionar ( tamanho = ( 200 , 400 ) )
Etapa 6: aplicar a transformação na imagem de entrada
Agora, aplique a transformação acima na imagem de entrada especificada para alterar sua dimensão:
new_img = transformar ( entrada_img )
Etapa 7: exibir imagem redimensionada e seu tamanho
Por fim, visualize a imagem redimensionada exibindo-a e verifique seu tamanho:
imprimir ( 'Novo tamanho da imagem após redimensionar:' , novo_img.tamanho )plt.imshow ( novo_img )
plt.show ( )
A saída abaixo indica que a imagem de entrada foi redimensionada com sucesso. Agora, sua largura é “400” e sua altura é “200”:
Da mesma forma, os usuários podem especificar qualquer outro tamanho para redimensionar a imagem. Aqui, redimensionaremos a mesma imagem de entrada com outro tamanho, ou seja, altura “250” e largura “150”:
transformar = Transformar.Redimensionar ( tamanho = ( 250 , 150 ) )
Isso redimensionará a imagem para novas dimensões:
Comparação
A comparação entre a imagem original e as imagens redimensionadas com dimensões diferentes pode ser vista abaixo:
Observação : Você pode acessar nosso Google Colab Notebook neste link .
Explicamos com eficiência o método para redimensionar uma imagem para o tamanho desejado no PyTorch.
Conclusão
Para redimensionar a imagem desejada para um tamanho específico no PyTorch, primeiro carregue a imagem desejada no Google Colab. Em seguida, importe as bibliotecas necessárias e leia a imagem de entrada. A seguir, calcule e imprima o tamanho da imagem de entrada. Depois disso, use o “ Redimensionar() ”Método para definir e aplicar a transformação na imagem de entrada desejada. Por fim, exiba a nova imagem redimensionada e seu tamanho. Este blog demonstrou o método para redimensionar a imagem desejada para um tamanho específico no PyTorch.