Skip to content

ChenHuaYou/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:

About

PyTorch implementation of DELG paper.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 92.5%
  • Python 7.5%