关于yolov5的om模型转换问题
Opened this issue · 0 comments
Jake-wei commented
我看过华为官网的几个yolov5模型的例子，不论是不是会添加华为为yolov5设计的后处理算子，在export_onnx时，都会修改yolo.py中的detect类的forward方法中代码， 添加一个self.training=True,我想请教一下，当前仓库中的yolov5在export_onnx时，会添加这一行代码吗
Opened this issue · 0 comments
我看过华为官网的几个yolov5模型的例子，不论是不是会添加华为为yolov5设计的后处理算子，在export_onnx时，都会修改yolo.py中的detect类的forward方法中代码， 添加一个self.training=True,我想请教一下，当前仓库中的yolov5在export_onnx时，会添加这一行代码吗