lyuyanyii / prototypical-network-pytorch

A re-implementation of "Prototypical Networks for Few-shot Learning"

Prototypical Network

A re-implementation of Prototypical Network.

Evaluated on Mini-ImageNet.

Results

1-shot: 49.1% (49.4% in paper)

5-shot: 66.9% (68.2% in paper)

Requirements

• python 3
• pytorch 0.4.0

Instructions

1. Download the images: https://drive.google.com/open?id=0B3Irx3uQNoBMQ1FlNXJsZUdYWEE

2. Make a folder materials/images and put those images into it.

3. mkdir save for saving models.

--gpu to specify device for program.

1-shot Train

python train.py

1-shot Test

python test.py

5-shot Train

python train.py --shot 5 --train-way 20 --save-path ./save/proto-5

5-shot Test

python test.py --load ./save/proto-5/max-acc.pth --shot 5