tensorflow-yolo
tensorflow-yolo copied to clipboard
Still get the former error, wierd!
Environment : TensorFlow 1.0 Python 3.6 (anaconda3)
I download the code for python3 version, and the pretrained model, and the VOC2007 data, then began to train the model by:
python tools/train.py -c conf/train.cfg
BUT, still get the error which I found in a closed issue (#11)
I also checked yolo_tiny_net.py, and found it has already been changed to:
predicts = tf.concat(3, [class_probs, scales, boxes])
Could you please give me some clue? Thanks a lot!
python 3 use tensorflow-0.11, and I haven't maintain "python 3" branch. I recommend you to use python2.7 version. (or you can modify all the different api in the codes manually according to https://www.tensorflow.org/install/migration)
@nilboy Thanks! I've changed the api and it runs ok.
Another question: How to use the model I've trained in the demo.py? I try to change the line below but it doesn't work:
#saver.restore(sess,'models/pretrain/yolo_tiny.ckpt')
saver.restore(sess,'models/train/model.ckpt-25000.data-00000-of-00001')
Maybe change to: saver.restore(sess,'models/train/model.ckpt-25000)
@xray1111 Would you be willing to share the updated code that is compatible with python3?