/tensorflow-yolov4-tflite

YOLOv4, YOLOv4-tiny, YOLOv3, YOLOv3-tiny Implemented in Tensorflow 2.0, Android. Convert YOLO v4 .weights tensorflow, tensorrt and tflite

Primary LanguagePythonMIT LicenseMIT

预训练权重文件yolov4.weights下载

百度网盘:https://pan.baidu.com/s/1gtSiJGe_RmYUbaiG8qGUlQ 提取码:abcd 下载后放入./data下

依赖库安装

pip install -r requirements-gpu.txt

数据集下载

cd ./script
./get_coco_dataset_2017.sh

训练命令

python train.py --weights ./data/yolov4.weights

验证命令

python evaluate.py --weights ./data/yolov4.weights
cd mAP/extra
python remove_space.py
cd ..
python main.py --output results_yolov4_tf

测试命令

源代码仓代码存在Bug，近期正在尝试修改

与标杆对比

无