본문 바로가기
Development/Onnx

[ONNX] Onnx-modifier

by shuka 2022. 10. 21.
728x90
728x90

Onnx-modifier란

모델의 구조를 visualize하기 위해 주로 사용하는 사이트가 netron.app이다.

onnx-modifier는 netron을 기반으로 개발됐고, onnx model을 visualization뿐만 아니라 editting까지 가능하다.

 

Onnx-modifier가 지원하는 기능

현재 지원하는 기능들은 다음과 같다.

  •  Delete/recover nodes
    • Delete a single node.
    • Delete a node and all the nodes rooted on it.
    • Recover a deleted node.
  •  Rename the node inputs/outputs
  •  Rename the model inputs/outputs
  •  Add new model outputs
  •  Edit the attribute of nodes
  •  Add new nodes (experimental)
  •  Change batch size

 

밑의 링크 보면 설명도 잘 되있고 생각보다 쓰기 편하게 되어 있어서 좋은거 같다.

GitHub - ZhangGe6/onnx-modifier: A tool to modify onnx models in a visualization fashion, based on Netron and Flask.

 

GitHub - ZhangGe6/onnx-modifier: A tool to modify onnx models in a visualization fashion, based on Netron and Flask.

A tool to modify onnx models in a visualization fashion, based on Netron and Flask. - GitHub - ZhangGe6/onnx-modifier: A tool to modify onnx models in a visualization fashion, based on Netron and F...

github.com

 

 

※출처: 꾸준희(https://eehoeskrap.tistory.com/)님 항상 좋은 정보 감사합니다.

728x90
반응형

댓글