参考博客:https://blog.csdn.net/thm225679/article/details/79407619
首先,下载yolo tensorflow及预先训练好的模型:
https://github.com/hizhangp/yolo_tensorflow
链接: https://pan.baidu.com/s/1htt9YBE 密码: ehw2
然后,在yolo_tensorflow-master文件夹下新建data文件夹,在data文件夹下新建weights子文件夹,并将训练好的模型放到weights文件夹下。
最后,在yolo_tensorflow-master文件夹下运行cmd,输入:
python test.py
由于安装tensorflow版本问题,可能会报错:
C:\Users\Administrator\Desktop\yolo_tensorflow-master>python test.py
Traceback (most recent call last):
File "test.py", line 211, in <module>
main()
File "test.py", line 197, in main
yolo = YOLONet(False)
File "C:\Users\Administrator\Desktop\yolo_tensorflow-master\yolo\yolo_net.py",
line 41, in __init__
is_training=is_training)
File "C:\Users\Administrator\Desktop\yolo_tensorflow-master\yolo\yolo_net.py",
line 69, in build_network
net, 64, 7, 2, padding='VALID', scope='conv_2')
File "d:\softW\Anaconda3\lib\site-packages\tensorflow\contrib\framework\python
\ops\arg_scope.py", line 181, in func_with_args
return func(*args, **current_args)
File "d:\softW\Anaconda3\lib\site-packages\tensorflow\contrib\layers\python\la
yers\layers.py", line 1039, in convolution
outputs = activation_fn(outputs)
File "C:\Users\Administrator\Desktop\yolo_tensorflow-master\yolo\yolo_net.py",
line 244, in op
return tf.nn.leaky_relu(inputs, alpha=alpha, name='leaky_relu')
AttributeError: module 'tensorflow.python.ops.nn' has no attribute 'leaky_relu'
tensorflow.python.ops.nnl没有eaky_relu属性,解决方法:
修改yolo_net.py文件的第242至245行
def leaky_relu(alpha):
def op(inputs):
return tf.nn.leaky_relu(inputs, alpha=alpha, name='leaky_relu')
return op
修改为:
def leaky_relu(alpha):
def op(inputs):
return tf.maximum(inputs, alpha * inputs)
return op
输出结果为:
如果想测试其他的图片,可以修改test.py文件的206行,修改为你要测试的图片路径即可。