zero-shot object detection

推荐搭配套餐:

  • Python 2.7
  • Keras 2.1.4
  • OpenCV 2.4.13
  • Tensorflow 1.3.0

或者:

  • Python 3.5-3.6
  • Keras 2.0.8
  • OpenCV 3.4.0
  • Tensorflow 1.3.0

 

安装:

1 本人套餐:

NVIDIA-SMI 384.130

CUDA Version 8.0.61

CUDNN_VERSION 6.0.21

pip install keras==2.1.4
pip install tensorflow-gpu==1.3.0
pip install opencv-python

 

2 下载h5格式的预训练文件,放入model文件夹:

https://www.dropbox.com/s/v6ueoa1g19bddao/model_frcnn.hdf5?dl=0

网页需要翻qiang下载,或者百度云下载:

链接: https://pan.baidu.com/s/1ZORTuXEoTNQHYkslLQW9vA    提取码: 5tui 

 

3 运行:

运行detect.py 批量检测Dataset/Sampleinput下的图片,输出图片在Dataset/Sampleoutput

 

问题 1:

ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory


Failed to load the native TensorFlow runtime.


解决:

这是因为我装了tensorflow-gpu 1.5版本,而我用的是cuda 8.0和cudnn6.0,1.5版本要求cuda 9.0,我的做法就是回滚:

 

pip install tensorflow-gpu==1.3.0

问题 2:

File "/home/thu/anaconda3/lib/python3.5/site-packages/tensorflow/python/framework/common_shapes.py", line 659, in _call_cpp_shape_fn_impl
    raise ValueError(err.message)
ValueError: Shape must be rank 1 but is rank 0 for 'bn_conv1/Reshape_4' (op: 'Reshape') with input shapes: [1,1,1,64], [].


解决:

pip install keras==2.1.4

 

问题 3:

OSError: Unable to open file (Unable to open file: name = 'model/model_frcnn.hdf5', errno = 2, error message = 'no such file or directory', flags = 0, o_flags = 0)


解决:

下载h5格式的预训练文件,放入model文件夹:

https://www.dropbox.com/s/v6ueoa1g19bddao/model_frcnn.hdf5?dl=0

网页需要翻qiang下载,或者百度云下载:

链接: https://pan.baidu.com/s/1ZORTuXEoTNQHYkslLQW9vA    提取码: 5tui 

 

 

 

Ref: https://github.com/salman-h-khan/ZSD_Release

 

你可能感兴趣的:(zero-shot)