dog_breed_classification_and_retrival

A project aimed to classify dog breeds and search for the most matched images in the database

The detail experiments and usage can be found in my website: https://bshoterj.github.io/2019/01/20/dog-retrival/

or visit my group member's github: https://github.com/INFINITSY/dog_test

Requirements

• Python3
• PyTorch
• keras
• dataset: Stanford dog dataset (18 classes)

Demo

Golden retrieve(金毛）

Siberian husky（哈士奇）

Dobeman(杜宾）

Model

• object detection: YOLOv3
• image retrival: ResNet50 + Euclid Distance