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.
Bài tập này là một phần của khóa học
Deep Learning cho Ảnh với PyTorch
Hướng dẫn bài tập
- Chuyển
bboxthành tensor bằngtorch.tensor(). - Thêm chiều batch để reshape
bbox_tensorbằngunsqueeze(0). - Tạo một transform để resize
imagevề(224)và chuyển thành một tensor ảnh chưa được scale. - Áp dụng
transformlênimage.
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
# 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)