.pb 파일에서 복원 된 모델의 입력 레이어 앞에 레이어를 추가하는 방법은 무엇입니까?

Question

필자는 .pb 모델 파일을 가지고 있으며 get_tensor_by_name을 사용하여 입력 텐서 및 출력 텐서를 얻은 다음 입력 및 출력 텐서를 사용하여 텐도 흐름 모델 서버에서 제공 할 수있는 저장된 모델을 만들 수 있습니다. 하지만 현재, 입력 텐서는 3D 배열 형식의 이미지이며 jpeg 인코딩 된 문자열을 디코딩하고 결과를 현재 입력 텐서로 공급하기 위해 하나 이상의 레이어를 추가하고 싶습니다. 어떻게해야할까요?

Answer 1

여기에서 파일 export_inference_graph를 편집하여 해당 레이어를 추가 할 수 있습니다. https://github.com/tensorflow/models/blob/master/research/slim/export_inference_graph.py#L114-L116.

자리 표시 자 코드 뒤에 추론 그래프를 내보내기 전에 원하는 모든 사전 처리 텐서 평가를 추가 할 수 있습니다.