yolov4初探学习_预测

概述地址https://github.comheAIGuysCodeensorflow-yolov4-tflite配置环境name:yolov4-gpudependencies:python==3.7pipmatplotlibopencvcudnncudatoolkit==10.1.243pip:tensorflow-gpu==2.3.0rc0opencv-python==4.1.1.26lxmltqdmabsl-pyeasydictpillow我的tens 地址

https://github.com/theAIGuysCode/tensorflow-yolov4-tflite

配置环境

name: yolov4-gpu

dependencIEs:

python==3.7pipmatplotlibopencvcudnncudatoolkit==10.1.243pip: tensorflow-gpu==2.3.0rc0opencv-python==4.1.1.26lxmltqdmabsl-pyeasydictpillow

我的tensorflow-gpu2.4.1 ，cuda10.0 能够正常使用。

下载weights文件

https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v4_pre/yolov4-tiny.weights

生成模型

python save_model.py --weights ./model/yolov4-tiny.weights --tiny true --output ./checkpoints/yolov4.tf --input_size 416 --model yolov4

进行预测

python detect.py --weights ./checkpoints/yolov4.tf --size 416 --model yolov4 --images kite.jpg --framework trt

总结

以上是内存溢出为你收集整理的yolov4初探学习_预测全部内容，希望文章能够帮你解决yolov4初探学习_预测所遇到的程序开发问题。

如果觉得内存溢出网站内容还不错，欢迎将内存溢出网站推荐给程序员好友。