이미지 텐서
한 커피 회사가 객체 감지 프로젝트를 진행하고 있으며, 관심 객체(여기서는 에스프레소 샷)에 주석을 달아야 해요. 에스프레소 샷 이미지에 대한 바운딩 박스 좌표 목록을 이미 만들어 두었어요. 이제 이미지와 좌표를 텐서로 변환해야 해요.
torch와 torchvision은 이미 임포트되어 있어요. torchvision.transforms는 transforms로 임포트되어 있어요. 이미지는 PIL 라이브러리의 Image.open()으로 불러와 image로 저장되어 있어요. 바운딩 박스 좌표는 변수 bbox에 저장되어 있어요.
이 연습은 강의의 일부입니다
PyTorch로 배우는 이미지 딥러닝
연습 안내
torch.tensor()를 사용해bbox를 텐서로 변환하세요.unsqueeze(0)으로 배치 차원을 추가해bbox_tensor의 형태를 바꾸세요.image를(224)로 리사이즈하고 스케일링하지 않은 이미지 텐서로 변환하는 변환기를 만드세요.image에transform을 적용하세요.
실습형 인터랙티브 연습
이 예제를 이 샘플 코드를 완성하여 풀어보세요.
# Convert bbox into tensors
bbox_tensor = ____
# Add a new batch dimension
bbox_tensor = bbox_tensor.____
# Resize image and transform tensor
transform = transforms.Compose([
transforms.____,
transforms.____
])
# Apply transform to image
image_tensor = ____
print(image_tensor)