
TensorFlow-YOLOv4-TFLite: YOLOv4, YOLOv4-tiny, YOLOv3, 和 YOLOv3-tiny 在 T...
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
简介:此项目实现YOLOv4、YOLOv4-tiny、YOLOv3及YOLOv3-tiny模型在TensorFlow框架下的转换至TFLite格式,适用于移动设备与嵌入式系统。
TensorFlow-YOLOv4-TFLite 是在 Tensorflow 2.0 中实现的 YOLO v4 和 YOLO v4-tiny 版本。该项目可以将YOLO v4、YOLO v3以及YOLO tiny 的.weights 文件转换为.pb、.tflite和trt格式,以便生成 TensorFlow、TensorFlow Lite 和 TensorRT。
性能演示版
# 将darknet权重转换成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/
全部评论 (0)
还没有任何评论哟~


