无法将YOLOv4转换为tflite

Question

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

我从得到转换错误

yolo.save_as_tflite("yolov4.tflite")

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

有人能推荐一些替代方法来转换tflite版本的yoo4吗(最好是在colab上工作(？

Answer 1

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

无论如何，为了获得Yolov4的tflite版本，我建议您使用此回购。

安装依赖项：

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权重保存到tflite转换所需的tfmodel：

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