我在将yolov5模型导出到tensorflow 时遇到了一些问题。我是tensorflow 和yolov5的新手,所以下面的文档给我带来了麻烦。什么是最简单的导出方法?
gzszwxb41#
Yolov5支持导出为两种TF格式:TensorFlow保存的模型和TensorFlow GraphDef。要导出到这两个模型:
python export.py --weights yolov5s.pt --include saved_model pb
用于推断
python detect.py --weights yolov5s_saved_model
和
python detect.py --weights yolov5s.pb
分别地
1条答案
按热度按时间gzszwxb41#
Yolov5支持导出为两种TF格式:TensorFlow保存的模型和TensorFlow GraphDef。要导出到这两个模型:
用于推断
和
分别地