Heim  >  Artikel  >  Betrieb und Instandhaltung  >  So lösen Sie das Problem der Installation von Caffe mit Conda und der Konvertierung von PB in Caffe unter Linux

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

PHPz
PHPznach vorne
2023-05-16 14:46:141467Durchsuche

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
So lösen Sie das Problem der Installation von Caffe mit Conda und der Konvertierung von PB in Caffe unter Linux4. 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
So lösen Sie das Problem der Installation von Caffe mit Conda und der Konvertierung von PB in Caffe unter Linux

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, EingabeformSo lösen Sie das Problem der Installation von Caffe mit Conda und der Konvertierung von PB in Caffe unter Linux

Merken Sie sich Name 7. Konvertieren Sie

pb ----> caffe

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

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!

Stellungnahme:
Dieser Artikel ist reproduziert unter:yisu.com. Bei Verstößen wenden Sie sich bitte an admin@php.cn löschen