tensorflow学习笔记(四):跑一跑yolo tensorflow

    参考博客:https://blog.csdn.net/thm225679/article/details/79407619

    首先,下载yolo tensorflow及预先训练好的模型:

https://github.com/hizhangp/yolo_tensorflow

   链接: pan.baidu.com/s/1htt9YB  密码: 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行,修改为你要测试的图片路径即可。

猜你喜欢

转载自blog.csdn.net/sinat_31425585/article/details/80902541