Praproses himpunan data gambar

Anda sedang mengembangkan sistem pertanian presisi untuk membantu petani memantau kesehatan tanaman, menggunakan model transformer yang telah dilatih sebelumnya, yang nantinya dapat Anda fine-tune pada citra pertanian. Praproses himpunan data menggunakan AutoImageProcessor untuk menyiapkannya bagi pelatihan!

Beberapa data telah dimuat sebelumnya:

Kelas AutoImageProcessor telah diimpor dari transformers
model bernilai microsoft/swin-tiny-patch4-window7-224
Sebuah dataset contoh telah didefinisikan, dengan gambar contoh yang dimuat ke dalam variabel image

Latihan ini adalah bagian dari kursus

Pelatihan Model AI Efisien dengan PyTorch

Lihat Kursus

Petunjuk latihan

Muat pemroses gambar yang telah dilatih sebelumnya dari model yang sudah ditentukan.
Petakan image_processor ke seluruh himpunan data.

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

# Load a pre-trained image processor
image_processor = ____.____(model)

# Map the image_processor to the entire dataset
dataset = dataset.____(
    lambda examples: {
        "pixel_values": [
            image_processor(image, return_tensors="pt").pixel_values
            for image in examples["img"]
        ]
    },
    batched=True,
)
print(dataset[0]["img"])

Edit dan Jalankan Kode