Comment résoudre le problème de l'installation de caffe avec conda et de la conversion de pb en caffe sous Linux

Démarrez directement depuis l'installation de caffe :

1. Créez d'abord un nouvel environnement python3.5

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

Entrez pour installer

2. Activez l'environnement

Une fois l'installation terminée, activez et entrez. l'environnement

source activate caffe-py3.5

3. Installez la version caffe-cpu ou la version caffe-gpu

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

4. Testez si l'installation est réussie :

python3import caffe

Aucune erreur n'indique que l'installation est réussie, quittez python

exit()

5. Installez tensorflow :

conda install -c defaults tensorflow==1.15.0

6. Installez mmdnn

pip install mmdnn

6. Nous prenons comme exemple le modèle mobilenetv1 téléchargé sur le site officiel de tf. Avant la conversion, utilisez netron pour afficher l'entrée et la sortie du modèle pb. nœuds nom, forme d'entrée :

Rappelez-vous nom et forme d'entrée

7 Convertir pb ---->caffe, entrée de ligne de commande :

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

8.

Converti avec succès en modèle caffe :

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!