YOLOv4 CrowdHuman 教程
这是一个演示如何使用和训练YOLOv4人检测器的。
目录
设置
如果您打算在上训练模型,您可以跳过本节并直接跳到上。
否则,要在本地运行训练,您需要有一台具有不错 GPU 的 x86_64 PC。 例如,我主要使用台式 PC 测试此存储库中的代码:
NVIDIA GeForce RTX 2080 Ti
Ubuntu 18.04.5 LTS (x86_64)
CUDA 10.2
cuDNN 8.0.1
此外,您应该在本地 PC 上正确安装 OpenCV(包括 python3“cv2”模块),因为数据准备代码和“darknet”都需要它。
准备训练数据
对于在本地 PC 上的训练,我使用“608x608”yolov4 模型作为示例。 请注意,我在本教程中只使用了 python3(python2 可能不起作用)。 请按照以下步骤准备“ Cr
1