Skip to content
Tensorflow code for ICML 2019 paper: LGM-Net: Learning to Generate Matching Networks for Few-Shot Learning
Branch: master
Clone or download
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
data remove extra empty lines May 12, 2019
saved_models Add files via upload May 11, 2019
LICENSE Initial commit May 11, 2019
README.md update README May 12, 2019
data.py remove extra empty lines May 12, 2019
experiment_builder.py
meta_matching_network.py remove extra empty lines May 12, 2019
storage.py remove extra empty lines May 12, 2019
train_meta_matching_network.py Update train_meta_matching_network.py May 31, 2019

README.md

LGM-Net

TensorFlow source code for the following publication:

LGM-Net: Learning to Generate Matching Networks for Few-Shot Learning

Huaiyu Li, Weiming Dong, Xing Mei, Chongyang Ma, Feiyue Huang, Bao-Gang Hu

In Proceedings of the 36th International Conference on Machine Learning (ICML 2019)

Requirements

Preparation

Set the path of resized miniImageNet dataset in data.py

Train

python train_meta_matching_network.py --way 5 --shot 1

python train_meta_matching_network.py --way 5 --shot 5

Test

python train_meta_matching_network.py --way 5 --shot 1 --is_test True --ckp checkpoint_id

Acknowledgements

Thanks to Antreas Antoniou for his Matching Networks implementation of which parts were used for this implementation.

You can’t perform that action at this time.