Tensor gambar

Sebuah perusahaan kopi memiliki proyek object detection yang memerlukan anotasi objek yang diminati, dalam hal ini, espresso shot. Anda telah membuat sebuah daftar berisi koordinat bounding box untuk sebuah gambar espresso shot. Sekarang, Anda perlu mengonversi gambar dan koordinat tersebut menjadi tensor.

torch dan torchvision telah diimpor. torchvision.transforms diimpor sebagai transforms. Gambar telah dimuat sebagai image menggunakan Image.open() dari pustaka PIL. Koordinat bounding box disimpan dalam variabel bbox.

espresso

Latihan ini adalah bagian dari kursus

Deep Learning untuk Gambar dengan PyTorch

Lihat Kursus

Petunjuk latihan

Konversikan bbox menjadi tensor menggunakan torch.tensor().
Ubah bentuk bbox_tensor dengan menambahkan dimensi batch menggunakan unsqueeze(0).
Buat transform untuk mengubah ukuran image menjadi (224) dan ubah menjadi tensor gambar tanpa penskalaan.
Terapkan transform pada image.

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

# 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)

Edit dan Jalankan Kode