无法将YOLOv4转换为tflite

Question

我正试图在我的android项目中使用yolo4，但在转换方面遇到了问题。 来自：https://pypi.org/project/yolov4/的代码在Google-Colab上运行良好，尽管它在Jupyter笔记本上的CUDA版本有一些问题

我从以下位置获得转换错误：

yolo.save_as_tflite("yolov4.tflite")

错误很长，我不确定我应该在这里粘贴哪个部分

有人能推荐一些替代方法来转换tflite版本的yolo4（最好是在colab上工作）

Answer 1

如果你已经发布了错误的重要部分，那会很有帮助。我猜您可能缺少一些依赖项/库

无论如何，要获得tflite版本的Yolov4，我建议您使用thisrepo

安装依赖项：

opencv-python==4.1.1.26
lxml
tqdm
tensorflow==2.3.0rc0
absl-py
easydict
matplotlib
pillow

下载yolov4.weights文件：https://drive.google.com/open?id=1cewMfusmPjYWbrnuJRuKhPMwRe_b9PaT

将darknet权重保存到tfmodel转换所需的tflite权重：

python save_model.py  weights ./data/yolov4.weights  output ./checkpoints/yolov4-416  input_size 416  model yolov4  framework tflite

转换为Yolov4tflite版本：

python convert_tflite.py  weights ./checkpoints/yolov4-416  output ./checkpoints/yolov4-416.tflite

此外，要测试转换的tflite模型，请执行以下操作：

python detect.py  weights ./checkpoints/yolov4-416.tflite  size 416  model yolov4  image ./data/kite.jpg  framework tflite