conda로 caffe를 설치하고 Linux에서 pb를 caffe로 변환하는 문제를 해결하는 방법

caffe 설치에서 직접 시작:

1. 먼저 새로운 python3.5 환경을 만듭니다.

conda create -n caffe-py3.5 python=3.5 -c defaults

설치하려면 y를 입력하세요.

2. 설치가 완료되면 활성화하고 들어갑니다. 환경

source activate caffe-py3.5

3. caffe-cpu 버전 또는 caffe-gpu 버전을 설치합니다.

conda install -c defaults caffeconda install -c defaults caffe-gpu

4. 설치가 성공했는지 테스트합니다.

python3import caffe

오류가 없으면 설치가 성공한 것입니다. Python

exit()

5을 종료합니다. tensorflow 설치:

conda install -c defaults tensorflow==1.15.0

6. mmdnn

pip install mmdnn

6을 설치합니다. 변환하기 전에

netron

을 사용하여 pb 모델 입력 및 출력을 봅니다. 노드 이름, 입력 모양:

이름 및

입력 모양

7을 기억하세요. pb ----> 명령줄 입력:

mmconvert -sf tensorflow -iw mobilenet_v1.pb --inNodeName input --inputShape 192,192,3 --dstNodeName MobilenetV1/Predictions/Reshape_1 -df caffe -om tf_mobilenet

8.

caffe 모델로 성공적으로 변환됨:

위 내용은 conda로 caffe를 설치하고 Linux에서 pb를 caffe로 변환하는 문제를 해결하는 방법의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!