1.Tensorflow-gpu环境搭建
环境搭建超级详细博客:TensorFlow2.1.0安装教程_YFCodeDream的博客-CSDN博客_tensorflow2.1
2.运行简介
1.输入:python convert.py生成tf可用的模型
2.识别
对图片进行预测,并且输出
python detect.py --image ./data/1.jpg
3.打开摄像头进行预测
python detect_video.py --video 0(0表示笔记本摄像头,1表示外机摄像头)
q关闭摄像头
4.对视频进行预测,并且输出
python detect_video.py --video test.mp4 --output ./test output.mp4
q关闭
注:上面是用命令运行也可用Pycharm
3.源码地址
百度网盘地址如下:
链接:https://pan.baidu.com/s/1JjfqoeA48I_jZRx_0-hLlQ
提取码:6pg7
GitHub地址如下:
GitHub - zzh8829/yolov3-tf2: YoloV3 Implemented in Tensorflow 2.0