tensorflow-yolov4-tflite
YOLOv4,YOLOv4-tiny在Tensorflow 2.0中实现。 将YOLO v4,YOLOv3,YOLO tiny .weights转换为.pb,.tflite和trt格式以生成tensorflow,tensorflow lite和tensorRT。
下载yolov4.weights文件: ://drive.google.com/open id 1cewMfusmPjYWbrnuJRuKhPMwRe_b9PaT
先决条件
Tensorflow 2.3.0rc0
性能
演示版
# Convert darknet weights to tensorflow
# # yolov4
python save_model.py --weights ./data/yolov4.weights --output ./checkpoints/yolov4-416 --input_size 416 --model yolov4
# # yolov4-tiny
python save_model.py --weights ./data/
1