ćwiczenie

Dostrajanie CV: klasy modelu

W tym ćwiczeniu wczytasz wstępnie wytrenowany model i dostosujesz jego wyjście do nowej klasyfikacji typów modeli samochodów ze zbioru danych Stanford Cars – zamiast 1000 klas używanych podczas oryginalnego trenowania na ImageNet. Zbiór danych zawiera otagowane obrazy samochodów.

Zbiór danych został wczytany (dataset), podobnie jak AutoModelForImageClassification z biblioteki transformers. Zbiór danych został przefiltrowany tak, aby uwzględniał trzy typy modeli.

Instrukcje

100 XP

Pobierz nowe nazwy etykiet ze zbioru danych.
Dodaj nowe mapowanie id2label podczas wczytywania modelu.
Dodaj odpowiadające mu mapowanie label2id.
Dodaj wymagany parametr, aby zmienić liczbę klas.

.css-6su6fj{-webkit-flex-shrink:0;-ms-flex-negative:0;flex-shrink:0;}ćwiczenie

Instrukcje

ćwiczenie