1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Deep Learning cho Ảnh với PyTorch
Connected

Bài tập

Tensor ảnh

Một công ty cà phê đang làm dự án nhận dạng đối tượng và cần gán nhãn các vật thể quan tâm, ở đây là các shot espresso. Bạn đã tạo một danh sách chứa toạ độ bounding box cho một ảnh espresso. Giờ, bạn cần chuyển đổi ảnh và các toạ độ này thành tensor.

torch và torchvision đã được nhập. torchvision.transforms được nhập với bí danh transforms. Ảnh đã được nạp vào biến image bằng Image.open() từ thư viện PIL. Các toạ độ bounding box được lưu trong biến bbox.

espresso

Hướng dẫn

100 XP
  • Chuyển bbox thành tensor bằng torch.tensor().
  • Thêm chiều batch để reshape bbox_tensor bằng unsqueeze(0).
  • Tạo một transform để resize image về (224) và chuyển thành một tensor ảnh chưa được scale.
  • Áp dụng transform lên image.