1、下载keras-yolov3:https://github.com/qqwweee/keras-yolo3
下载yolov3.weights:https://pjreddie.com/media/files/yolov3.weights将其放在keras-yolo3目录下
然后在keras-yolo3文件目录运行一下命令,得到yolo.h5
python convert.py yolov3.cfg yolov3.weights model_data/yolo.h5 (中间要安装Numpy、TensorFlow、Keras、OpenCV、Pillow等环境)
2、下载deep_sort:https://github.com/nwojke/deep_sort
将keras-yolo3目录下的model_data文件夹里面的yolo.h5和mars-small128.pb复制到deep_sort目录下的model_data文件夹里面
3、运行deep_sort里面的yolo.py会报错:module 'keras.backend' has no attribute 'get_session'是由于tensorflow版本的问题
解决办法:将yolo.py里面44行修改为以下内容:
4、运行deep_sort里面的yolo_video.py 。会报一个读取video_path文件错误(因为没有合适的视频文件),将其改为从笔记本摄像头读取。
5、再次运行报错:TypeError: Tensor is unhashable if Tensor equality is enabled. Instead, use tensor.experimental_r 也是版本问题
解决办法:在添加以下代码,就可以啦!
import tensorflow tensorflow.compat.v1.disable_v2_behavior()