上传者: 42131352
|
上传时间: 2022-02-11 08:53:49
|
文件大小: 91.75MB
|
文件类型: -
TensorFlow-2.x-YOLOv3和YOLOv4教程
TensorFlow 2.x中的YOLOv3和YOLOv4实现,并支持训练,传递训练,对象跟踪mAP等...代码已通过以下规格测试:
i7-7700k CPU和Nvidia 1080TI GPU
操作系统Ubuntu 18.04
CUDA 10.1
cuDNN v7.6.5
TensorRT-6.0.1.5
Tensorflow-GPU 2.3.1
代码已在Ubuntu和Windows 10上进行了测试(不正式支持TensorRT)
使用2.4.0 TensorFlow-不起作用!
安装
首先,克隆或下载此GitHub存储库。安装要求并下载预训练的砝码:
pip install -r ./requirements.txt
# yolov3
wget -P model_data https://pjreddie.com