1. 학습
    2. /
  2. 강의
    3. /
  3. PyTorch로 배우는 이미지 딥러닝
Connected

연습 문제

바운딩 박스 그리기

이미지와 바운딩 박스 텐서를 준비했으니, 이제 이미지 위에 박스를 그려 정확도를 눈으로 확인해 보세요.

torch, torchvision, torchvision.transforms는 이미 임포트되어 있어요. 이미지는 image_tensor로 텐서 형태로 변환되어 있고, 좌표는 x_min, y_min, x_max, y_max 변수에 할당되어 있어요.

지침

100 XP
  • torchvision.utils에서 draw_bounding_boxes를 임포트하세요.
  • x_min, y_min, x_max, y_max로 이루어진 리스트로 바운딩 박스 bbox를 정의하세요.
  • 이미지 위에 박스를 그리기 위해 image_tensor와 bbox_tensor를 draw_bounding_boxes에 전달하고, 출력 결과를 img_bbox에 할당하세요.
  • img_bbox 텐서를 이미지로 변환하세요.