smallcorgi / Faster-RCNN_TF

Faster-RCNN in Tensorflow
MIT License
2.34k stars 1.12k forks source link

con not get .ckpt file after training #233

Open longgang123 opened 6 years ago

longgang123 commented 6 years ago

command use this ./experiments/scripts/faster_rcnn_end2end.sh GPU 0 VGG16 pascal_voc while end training it shows like this below: can anybody know the reason? speed: 0.205s / iter Wrote snapshot to: /home/alex/software/Faster-RCNN_TF/output/faster_rcnn_end2end/voc_2007_trainval/VGGnet_fast_rcnn_iter_700.ckpt done solving

real 7m5.121s user 5m46.348s sys 0m46.364s

zzzzzz0407 commented 6 years ago

I have the same issue! And i just find the ckpt in appointed dir!

shaohuifangit commented 6 years ago

I have the same issue! How did you solve it?

zzzzzz0407 commented 6 years ago

You have to specify write_version=tf.train.SaverDef.V1 in /lib/fast_rcnn/train.py, if you're using tensorflow version lower than 1.0, to get .ckpt file. just like this: saver = tf.train.Saver(max_to_keep=100, write_version=tf.train.SaverDef.V1)

shaohuifangit commented 6 years ago

But my tensorflow version is 1.3.

IMAUZS commented 6 years ago

i'm also meeting the same problem.i used tensorflow-1.1,tensorflow-1.2,tensroflow-1.3.there 's still the same problem.after i run train,there can not get the cpkt file . when i modified the train_net.py to saver = tf.train.Saver(max_to_keep=100, write_version=tf.train.SaverDef.V1),there shows a error that "AttributeError: 'module' object has no attribute 'Save'".i don't know why it happens. there's anyone gives more infos to solve this problem? thank you.

YCIrving commented 6 years ago

79 can solve this problem though you can just comment the loop in test_net.py like #241