How to load tensorflow's new v2 format checkpoint?

CharlesShang / TFFRCNN

FastER RCNN built on tensorflow

MIT License

874 stars 418 forks source link

How to load tensorflow's new v2 format checkpoint? #128

Open SignalImageCV opened 6 years ago

SignalImageCV commented 6 years ago

Hi, the author's test_net code maybe just applicable for v1 format checkpoint, but after tensorflow r1.2, the checkpoint will be saved as v2 format by default. How to save into v2 format and restore v2 format? not v1, thanks!

allanwlz commented 6 years ago

aver.restore(sess, args.model) Use args.model = '.ckpt', instead of '.ckpt.meta'