ONNX 모델에 출력 노드를 추가하는 방법
import onnx
# モデルの出力ファイル名
model_path = "path to model"
model = onnx.load(model_path)
input_path = 'path to load model'
output_path = 'path to save model'
input_names = ['input_1']
output_names = ['Logits_softmax', 'Logits_softmax_Flatten'] # 左が元々あったノード 、 右が追加するノード
onnx.utils.extract_model(input_path, output_path, input_names, output_names)
핑크의 원이 추가한 출력 노드가 됩니다.
이제 Softmax 앞의 출력을 얻을 수 있습니다.
Reference
이 문제에 관하여(ONNX 모델에 출력 노드를 추가하는 방법), 우리는 이곳에서 더 많은 자료를 발견하고 링크를 클릭하여 보았다 https://qiita.com/yudai1102jp/items/3a989eaf048d362d021b텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
우수한 개발자 콘텐츠 발견에 전념
(Collection and Share based on the CC Protocol.)
좋은 웹페이지 즐겨찾기
