1. Lära sig
    2. /
  2. Courses
    3. /
  3. PyTorch로 배우는 이미지 딥러닝
Connected

exercise

이미지 텐서

한 커피 회사가 객체 감지 프로젝트를 진행하고 있으며, 관심 객체(여기서는 에스프레소 샷)에 주석을 달아야 해요. 에스프레소 샷 이미지에 대한 바운딩 박스 좌표 목록을 이미 만들어 두었어요. 이제 이미지와 좌표를 텐서로 변환해야 해요.

torch와 torchvision은 이미 임포트되어 있어요. torchvision.transforms는 transforms로 임포트되어 있어요. 이미지는 PIL 라이브러리의 Image.open()으로 불러와 image로 저장되어 있어요. 바운딩 박스 좌표는 변수 bbox에 저장되어 있어요.

espresso

Instruktioner

100 XP
  • torch.tensor()를 사용해 bbox를 텐서로 변환하세요.
  • unsqueeze(0)으로 배치 차원을 추가해 bbox_tensor의 형태를 바꾸세요.
  • image를 (224)로 리사이즈하고 스케일링하지 않은 이미지 텐서로 변환하는 변환기를 만드세요.
  • image에 transform을 적용하세요.