Few thoughts on the tool from Titus' feedback!
- Train on multiple images
- Load pretrained model for classification on new images (export function already exists)
- Iterative improvement over images (use features from previous image for the next ones to improve classification)