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
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.53. Installieren Sie die Caffe-CPU-Version oder die Caffe-GPU-Version
conda install -c defaults caffeconda install -c defaults caffe-gpu4. Testen Sie, ob die Installation erfolgreich ist:
python3import caffeWenn 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 mmdnn6. Wir nehmen das von der offiziellen TF-Website heruntergeladene
netv1
. Verwenden Sie vor der Konvertierungnetron
, um das pb-Modell anzuzeigen Eingabe- und AusgabeknotenName, 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_mobilenet8. 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!