我试图在我的安卓项目中使用yolo4
,但我在转换方面遇到了问题。来自:https://pypi.org/project/yolov4/的代码在Google-Colab
上运行得很好,尽管它在Jupyter
笔记本上的CUDA
版本有一些问题。
我从以下方面得到了转换错误:
yolo.save_as_tflite("yolov4.tflite")
错误很长,我不确定我应该粘贴在这里的哪个部分。
有人能推荐一些替代的方法来转换tflite
版本的yolo4 (最好是使用colab
)吗?
发布于 2020-07-19 22:20:46
如果您已经发布了错误的重要部分,这将是有帮助的。我猜您可能缺少了一些依赖关系/库。
无论如何,要获得tflite
版本的Yolov4
,我建议您使用这回购。
安装依赖项:
opencv-python==4.1.1.26
lxml
tqdm
tensorflow==2.3.0rc0
absl-py
easydict
matplotlib
pillow
下载yolov4.权重文件: b9PaT
将darknet
权重保存到tfmodel
中,这是tflite
转换所需的:
python save_model.py --weights ./data/yolov4.weights --output ./checkpoints/yolov4-416 --input_size 416 --model yolov4 --framework tflite
转换为Yolov4 tflite
版本:
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
https://stackoverflow.com/questions/62968981
复制