[ONNX] ONNX 변환모델에 메타데이터 추가하기

이번 포스팅에서는 ONNX 변환 모델에 메타데이터를 추가하는 방법에 대해 설명하겠습니다.

변환할 모델에 여러 정보를 추가하는 방법은 metadata_probs.add() 함수를 이용하면 됩니다!

위 함수를 이용해 ONNX 모델에 변환날짜와 변환장소를 추가해보겠습니다. 

 

먼저 사용할 ONNX 모델을 로드합니다.

onnx_model = onnx.load('my_onnx.onnx')

 

이제 메타 데이터를 추가해보겠습니다. 아래와 같이 여러 정보들에 대해 key와 value를 각각 설정해줄 수 있습니다.

meta_date = onnx_model.metadata_props.add()
meta_date.key = "date"
meta_date.value = "2023/04/01"

meta_place = onnx_model.metadata_props.add()
meta_place.key = "place"
meta_place.value = "starbucks"

 

메타데이터를 추가하여 ONNX 모델을 다시 저장해줍니다.

onnx.save(onnx_model, "my_onnx.onnx")