tzl1

你好我想请教下， 如何将.h5的权重换成.onnx的呢

1

以下是我做的一些尝试，但是都失败了： import keras2onnx,keras import onnx from nets.yolo4 import yolo_body from keras.models import load_model num_anchors = 9 num_classes = 12 model_path = r'E:/python_pro/test/model_data/last1.h5' model = yolo_body(Input(shape=(None,None,3)), num_anchors//3, num_classes) model.load_weights(model_path) onnx_model =...

你好我想问下训练完以后怎么测试呢