pytorch 将Yolov5导出到Tensorflow

dly7yett  于 2022-11-09  发布在  其他
关注(0)|答案(1)|浏览(217)

我在将yolov5模型导出到tensorflow 时遇到了一些问题。我是tensorflow 和yolov5的新手,所以下面的文档给我带来了麻烦。什么是最简单的导出方法?

gzszwxb4

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

分别地

相关问题