tensorflow-yolo icon indicating copy to clipboard operation
tensorflow-yolo copied to clipboard

Still get the former error, wierd!

Open xray1111 opened this issue 7 years ago • 4 comments

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) image

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!

xray1111 avatar Apr 26 '17 16:04 xray1111

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 avatar Apr 27 '17 01:04 nilboy

@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')

xray1111 avatar Apr 27 '17 03:04 xray1111

Maybe change to: saver.restore(sess,'models/train/model.ckpt-25000)

nilboy avatar Apr 27 '17 03:04 nilboy

@xray1111 Would you be willing to share the updated code that is compatible with python3?

salilkapur avatar May 26 '17 16:05 salilkapur