PyTorch는 수많은 이미지 작업을 위한 다양한 도구와 모듈을 제공하는 잘 알려진 딥 러닝 프레임워크입니다. 또한 자르기, 크기 조정, 회전, 뒤집기, 크기 조정 등과 같은 이미지의 다양한 변환을 수행하는 클래스와 함수 세트가 있는 'torchvision.transforms' 모듈을 제공합니다. 사용자는 “ 크기 조정() ” 메서드를 사용하여 크기, 즉 높이와 너비를 지정하여 원하는 이미지의 크기를 조정합니다. 이 메서드는 지정된 크기의 크기가 조정된 새 이미지를 반환합니다.
이 블로그에서는 PyTorch에서 이미지 크기를 특정 크기로 조정하는 방법을 보여줍니다.
PyTorch에서 이미지 크기를 특정 크기로 조정하는 방법은 무엇입니까?
PyTorch에서 이미지의 크기를 변경하고 특정 크기로 크기를 조정하려면 아래 나열된 단계를 따르세요.
- Google Colab에 이미지 업로드
- 필수 라이브러리 가져오기
- 입력 이미지 읽기
- 입력 이미지 크기 계산 및 인쇄
- 변환 만들기
- 지정된 이미지에 변환 적용
- 크기가 조정된 이미지와 해당 크기를 표시합니다.
1단계: Google Colab에 이미지 업로드
먼저 Google Colab을 열고 아래 강조 표시된 아이콘을 클릭하세요. 그런 다음 컴퓨터에서 특정 이미지를 선택하고 업로드합니다.
그 후 이미지가 Google Colab에 업로드됩니다.
여기서는 다음 이미지를 업로드했으며 특정 크기로 크기를 조정합니다.
2단계: 필수 라이브러리 가져오기
다음으로 필요한 라이브러리를 가져옵니다. 예를 들어 다음 라이브러리를 가져왔습니다.
수입 토치torchvision.transforms 가져오기 ~처럼 변환
PIL 가져오기 이미지에서
matplotlib.pyplot 가져오기 ~처럼 plt
여기:
- “ 수입 토치 ”는 PyTorch 라이브러리를 가져옵니다.
- “ torchvision.transforms를 변환으로 가져오기 ”는 이미지 데이터를 신경망에 공급하기 전에 전처리하는 데 사용되는 torchvision의 변환 모듈을 가져옵니다.
- “ PIL 가져오기 이미지에서 ”는 다양한 이미지 파일 형식을 열고 저장합니다.
- “ matplotlib.pyplot을 plt로 가져오기 ”는 시각화 및 플롯을 생성하는 데 사용되는 “pyplot” 모듈을 가져옵니다.
3단계: 입력 이미지 읽기
그런 다음 컴퓨터에서 입력 이미지를 읽습니다. 여기서 우리는 “ galaxy_img.jpg ”를 입력하고 이를 “ input_img ” 변수:
input_img=이미지.열기 ( 'galaxy_img.jpg' )
4단계: 입력 이미지 크기 계산 및 인쇄
그런 다음 입력 이미지의 크기, 즉 너비와 높이를 계산하고 인쇄합니다.
크기 = input_img.size인쇄 ( '원본(입력) 이미지 크기:' , 크기 )
입력 이미지의 너비는 '384'이고 이미지의 높이는 '576'임을 확인할 수 있습니다.
5단계: 변환 생성
다음으로, 입력 이미지의 크기를 조정하는 변환을 정의합니다. 사용자는 새 이미지의 높이와 너비 등 새 크기를 지정해야 합니다. 여기서는 높이 '200'과 너비 '400'을 지정했습니다.
변환 = 변환.크기 조정 ( 크기 = ( 200 , 400 ) )
6단계: 입력 이미지에 변환 적용
이제 지정된 입력 이미지에 위의 변환을 적용하여 차원을 변경합니다.
new_img = 변환 ( input_img )
7단계: 크기가 조정된 이미지 및 해당 크기 표시
마지막으로 크기가 조정된 이미지를 표시하여 보고 크기를 확인합니다.
인쇄 ( '크기 조정 후 이미지의 새 크기:' , new_img.size )plt.imshow ( new_img )
plt.show ( )
아래 출력은 입력 이미지의 크기가 성공적으로 조정되었음을 나타냅니다. 이제 너비는 '400'이고 높이는 '200'입니다.
마찬가지로 사용자는 다른 크기를 지정하여 이미지 크기를 조정할 수 있습니다. 여기서는 동일한 입력 이미지의 크기를 다른 크기(예: 높이 '250' 및 너비 '150')로 조정합니다.
변환 = 변환.크기 조정 ( 크기 = ( 250 , 150 ) )
그러면 이미지 크기가 새로운 크기로 조정됩니다.
비교
원본 이미지와 크기가 다른 크기로 조정된 이미지를 비교하면 아래에서 볼 수 있습니다.
메모 : 여기에서 Google Colab Notebook에 액세스할 수 있습니다. 링크 .
PyTorch에서 이미지를 원하는 크기로 조정하는 방법을 효율적으로 설명했습니다.
결론
PyTorch에서 원하는 이미지의 크기를 특정 크기로 조정하려면 먼저 원하는 이미지를 Google Colab에 업로드하세요. 그런 다음 필요한 라이브러리를 가져오고 입력 이미지를 읽습니다. 다음으로 입력 이미지의 크기를 계산하고 인쇄합니다. 그 후에는 ' 크기 조정() ” 메서드를 사용하여 원하는 입력 이미지에 변환을 정의하고 적용합니다. 마지막으로 크기가 조정된 새 이미지와 해당 크기를 표시합니다. 이 블로그에서는 PyTorch에서 원하는 이미지의 크기를 특정 크기로 조정하는 방법을 시연했습니다.