Skip to content

ditwoo/pytorch-delg-example

Repository files navigation

Data

Used small subset of ImageNet - Imagewoof, dataset with 10 classes of dogs.

Training process

From paper above it's important to train model in two steps - training global features & training local features.

Global features produce embedding like this:

(Big black points - class centroids)

For training local features used attention, here are few examples of attention maps: