YOLOv3(Pytorch版本和Tensorflow版本)学习

一、地址来源

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

你可能感兴趣的:(机器学习,pytorch,tensorflow,深度学习)