In vier stappen van foto's naar een getest model

1Dataset maken

Maak een dataset en geef de dingen een naam die het model moet leren herkennen, zoals 'bout' of 'kras'. Dat zijn je klassen. Daarna markeer je ze op je foto's: met een bounding box om elk object, met de precieze omtrek (polygoon), of met keypoints op vaste plekken ervan. Bij een omtrek helpt SAM 2: je klikt één keer op het object en de app tekent de omtrek, die je daarna kunt bijstellen.

Je krijgt een gelabelde set foto's, die automatisch wordt verdeeld in foto's om op te trainen en foto's om mee te testen.

• Auto-label. Heb je eenmaal een getraind model, dan kan dat nieuwe foto's voor je labelen. De voorstellen komen als concept in het labelscherm; jij controleert ze, verbetert wat niet klopt en slaat op.
• Augmentatie. Voeg gespiegelde, gedraaide of lichtere kopieën van je foto's toe, zodat het model van meer voorbeelden leert.
• Auto-balance. Heeft één klasse veel minder voorbeelden dan de rest, dan voegt de app daar extra uitsneden van toe aan de trainingsfoto's.
• Label Studio. Label je al in Label Studio, een los labelprogramma? Dan kun je die labels importeren.

2Model trainen

Kies je dataset en een modelgrootte en druk op Training starten. De training draait op je eigen GPU. Je kunt hem stoppen en later verdergaan waar hij gebleven was.

Je krijgt een getraind model, met een score en een grafiek die laten zien hoe goed het leert. Een ingebouwde rondleiding legt uit wat die score betekent.

• Drie soorten modellen. Het ene vindt objecten en zet er een bounding box omheen (Faster R-CNN). Het tweede zet er ook keypoints op (Keypoint R-CNN). Het derde volgt de precieze omtrek van elk object (Mask R-CNN).
• Drie groottes. Een groter model kan nauwkeuriger zijn, maar heeft meer VRAM nodig. Hoeveel, zie je voordat je begint.

3Model testen

Laat het model los op testfoto's die het tijdens de training niet heeft gezien, of op je eigen foto's.

Je krijgt de foto's met wat het model heeft gevonden, een score per klasse en een confusion matrix: een tabel die laat zien welke klassen het model met elkaar verwart, met de grootste fouten eronder op een rij. Modellen met omtrekken worden beoordeeld op die omtrek, niet alleen op een kader.

4Modellen vergelijken

Al je geteste modellen staan in één ranglijst, de beste bovenaan. Kies er twee, bijvoorbeeld een van vóór en een van na het toevoegen van extra foto's.

Je ziet hun scores en resultaten naast elkaar, dus je ziet meteen welke beter is. Die houd je.