So lösen Sie das Problem der Installation von Caffe mit Conda und der Konvertierung von PB in Caffe unter Linux-Betrieb und Wartung von Linux-php.cn

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

PHPz

Freigeben： 2023-05-16 14:46:14

nach vorne

1503 Leute haben es durchsucht

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

Nach dem Login kopieren

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

source activate caffe-py3.5

Nach dem Login kopieren

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

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

Nach dem Login kopieren

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

4. Testen Sie, ob die Installation erfolgreich ist:

python3import caffe

Nach dem Login kopieren

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

exit()

Nach dem Login kopieren

5. Installieren Sie tensorflow:

conda install -c defaults tensorflow==1.15.0

Nach dem Login kopieren

pip install mmdnn

Nach dem Login kopieren

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 So 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

Nach dem Login kopieren

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!