File size: 545 Bytes
88359db |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
import onnx
from onnx import compose
# Load the models
model1 = onnx.load("preprocessing.onnx")
model2 = onnx.load("eva02_large_patch14_448.onnx")
# Merge the models
merged_model = compose.merge_models(
model1,
model2,
io_map=[("output_preprocessing", "input")],
prefix1="preprocessing_",
prefix2="model_",
doc_string="Merged preprocessing and eva02_large_patch14_448 model",
producer_name="[email protected] using onnx compose",
)
# Save the merged model
onnx.save(merged_model, "merged_model_compose.onnx")
|