web-dev-qa-db-fra.com

Définir GLOG_minloglevel = 1 pour empêcher la sortie dans Shell de Caffe

J'utilise Caffe, qui imprime beaucoup de sortie vers le Shell lors du chargement du réseau neuronal.
Je voudrais supprimer cette sortie, ce qui peut être fait en définissant GLOG_minloglevel=1 lors de l'exécution du script Python. J'ai essayé de le faire en utilisant le code suivant, mais j'obtiens toujours toute la sortie du chargement du net. Comment supprimer correctement la sortie?

os.environ["GLOG_minloglevel"] = "1"
net = caffe.Net(model_file, pretrained, caffe.TEST)
os.environ["GLOG_minloglevel"] = "0"
pythonneural-networkdeep-learningcaffeglog
25
pir

Pour supprimer le niveau de sortie, vous devez augmenter le niveau de journalisation à au moins 2

 os.environ['GLOG_minloglevel'] = '2'

Les niveaux sont

0 - débogage
1 - info (encore BEAUCOUP de sorties)
2 - avertissements
3 - erreurs

Mise à jour:
Comme cet indicateur est global sur caffe, il doit être défini prior sur l'importation de caffe package (comme indiqué par jbum ). Une fois l'indicateur défini et caffe importé, le comportement de l'outil GLOG ne peut pas être modifié.

47
Shai

J'ai pu faire fonctionner la solution de Shai , mais seulement en exécutant cette ligne dans Python avant appeler

import caffe
28
jbum