一、地址来源

YOLOv4最全复现代码合集（含PyTorch/TF/Keras和Caffe等）

二、Pytorch版本

地址：https://github.com/Tianxiaomo/pytorch-YOLOv4

这个地址支持训练

Requirements and Dependencies

pip install numpy==1.18.2
# CPU only
pip install torch==1.4.0+cpu torchvision==0.5.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

pip install tensorboardX==2.0 
pip install scikit_image==0.16.2 
pip install matplotlib==2.2.3 
pip install tqdm==4.43.0 
pip install easydict==1.9 
pip install Pillow==7.1.2 
pip install skimage 
pip install opencv_python==4.1.2.30
pip install pycocotools

三、Tensorflow版本

地址：GitHub – hunglc007/tensorflow-yolov4-tflite: YOLOv4, YOLOv4-tiny, YOLOv3, YOLOv3-tiny Implemented in Tensorflow 2.0, Android. Convert YOLO v4 .weights tensorflow, tensorrt and tflite

这个支持训练

Requirements and Dependencies

https://download.pytorch.org/whl/torch_stable.html
pip install scikit_image==0.16.2 
pip install tensorflow==2.3.0
pip install easydict==1.9 
pip install Pillow==7.1.2 
pip install opencv_python==4.1.2.30