search

nilboy / tensorflow-yolo

tensorflow implementation of 'YOLO : Real-Time Object Detection'(train and test)
774 stars 312 forks source link

Training Run-time Error #11

Closed aliceyayunji closed 7 years ago

aliceyayunji commented 7 years ago

I downloaded the code and the tiny yolo pretrain model, change nothing but Error Occurs. Environment : TensorFlow 1.0 GPU Titan X Python 2.7

image

When change the config to use YoloNet there's no such problem, but the pretrain model doesn't fit ...

nilboy commented 7 years ago

Please make sure you use the correct tensorflow version(1.0), because the tf.concat api has changed from tf-1.0. you can reference to http://stackoverflow.com/questions/41813665/tensorflow-slim-typeerror-expected-int32-got-list-containing-tensors-of-type

aliceyayunji commented 7 years ago

It is the problem about the tensorflow version. Reference to the link nilboy supply, I solved the problem: just change the code in yolo_tiny_net.py :

predicts = tf.concat([class_probs, scales, boxes], 3)

to predicts = tf.concat(3,[class_probs, scales, boxes]) and it can run ok.