Předzpracování datových sad s obrázky

Vyvíjíš systém přesného zemědělství, který pomáhá zemědělcům sledovat zdraví plodin. Využíváš předtrénovaný transformerový model, který později doladíš na zemědělských snímcích. Předzpracuj datovou sadu pomocí AutoImageProcessor a připrav ji na trénování!

Některá data jsou předpřipravena:

Třída AutoImageProcessor je importována z transformers
model je nastaveno na microsoft/swin-tiny-patch4-window7-224
Je definována ukázková dataset s ukázkovým obrázkem načteným do proměnné image

Toto cvičení je součástí kurzu

Efficient AI Model Training with PyTorch

Zobrazit kurz

Pokyny k cvičení

Načti předtrénovaný procesor obrázků z předem definovaného model.
Namapuj image_processor na celou datovou sadu.

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

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

Upravit a spustit kód