天天看點

目标檢測_YOLO V4YOLOV4

YOLOV4

Paper:https://arxiv.org/abs/2004.10934

GitHub:https://github.com/AlexeyAB/darknet

跑通官方代碼:

# 建立虛拟環境
conda create -n py37_yolo python=3.7
source activate py37_yolo
# 安裝 opencv
pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple
# 下載下傳官方代碼
git clone https://github.com/AlexeyAB/darknet.git
# 修改編譯檔案 Makefile
cd darknet/
vim Makefile
# 修改 Makefile 中的字段如下
GPU=1 CUDNN=1 CUDNN_HALF=1 OPENCV=1
# 編譯
cd darknet/
make
# 下載下傳官方權重 yolov4 weights
連結: https://pan.baidu.com/s/1cXOJ80qpcg7nTXYpHjVd1g  
密碼: 0c96
# 把 yolov4.weights 放在 darknet/ 下
mv yolov4.weights darknet/
# 把測試視訊放在 darknet/data/ 下
mv test_3.mp4 darknet/data
# 測試 yolov4.weights
cd darknet/
./darknet detector demo cfg/coco.data cfg/yolov4.cfg yolov4.weights data/test_3.mp4 -ext_output

           

繼續閱讀