So lösen Sie das Problem der Installation von Caffe mit Conda und der Konvertierung von PB in Caffe unter Linux

Starten Sie direkt mit der Installation von caffe:

1. Erstellen Sie zunächst eine neue Python3.5-Umgebung.

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

. Geben Sie y ein, um die Umgebung zu installieren die Umgebung

source activate caffe-py3.5

3. Installieren Sie die Caffe-CPU-Version oder die Caffe-GPU-Version

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

4. Testen Sie, ob die Installation erfolgreich ist:

python3import caffe

Wenn kein Fehler gemeldet wird, ist die Installation erfolgreich und beenden Sie Python

exit()

5. Installieren Sie tensorflow:

conda install -c defaults tensorflow==1.15.0

pip install mmdnn

6. Wir nehmen das von der offiziellen TF-Website heruntergeladene

netv1

. Verwenden Sie vor der Konvertierung

netron

, um das pb-Modell anzuzeigen Eingabe- und Ausgabeknoten

Name, Eingabeform

Merken Sie sich Name 7. Konvertieren Sie

pb ----> caffe

, Befehlszeileneingabe:

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

8. Konvertierungsergebnis:

Erfolgreich in das Caffe-Modell konvertiert:

Das obige ist der detaillierte Inhalt vonSo lösen Sie das Problem der Installation von Caffe mit Conda und der Konvertierung von PB in Caffe unter Linux. Für weitere Informationen folgen Sie bitte anderen verwandten Artikeln auf der PHP chinesischen Website!