Skip to content
master
Switch branches/tags
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 
 
 

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

Releases

No releases published

Packages

No packages published

Languages