Bagaimana untuk menyelesaikan masalah memasang caffe dengan conda dan menukar pb kepada caffe di bawah Linux

Mulakan terus dari pemasangan kafe:

1 Mula-mula buat persekitaran python3.5 baharu

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

Masukkan y untuk memasang

2. Aktifkan persekitaran

Selepas pemasangan, aktifkan dan masukkan persekitaran

source activate caffe-py3.5

3 Pasang versi caffe-cpu atau versi caffe-gpu

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

4. Uji sama ada pemasangan berjaya:

python3import caffe

Jika tiada ralat dilaporkan, pemasangan berjaya dan keluar dari python

exit()

5 aliran tensor:

conda install -c defaults tensorflow==1.15.0

6 Pemasangan mmdnn

pip install mmdnn

6 Kami mengambil model mobilenetv1 yang dimuat turun dari tapak web rasmi tf. contoh. Sebelum penukaran, gunakan netron untuk melihat model pb nama, bentuk input :

Ingat. nama dan bentuk input

7 Tukar pb ----> input baris arahan: rreee

8. Hasil penukaran:

Berjaya ditukar kepada model kafe:

Atas ialah kandungan terperinci Bagaimana untuk menyelesaikan masalah memasang caffe dengan conda dan menukar pb kepada caffe di bawah Linux. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!