Skip to content

Tensorflow code for ICML 2019 paper: LGM-Net: Learning to Generate Matching Networks for Few-Shot Learning

License

Notifications You must be signed in to change notification settings

likesiwell/LGM-Net

Repository files navigation

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.

About

Tensorflow code for ICML 2019 paper: LGM-Net: Learning to Generate Matching Networks for Few-Shot Learning

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages