2015-04-22 11 views
19

Używam Caffe, który drukuje wiele danych wyjściowych do powłoki podczas ładowania sieci neuronowej.
Chciałbym wyłączyć to wyjście, co można zrobić, ustawiając GLOG_minloglevel=1 podczas uruchamiania skryptu Python. Próbowałem to zrobić za pomocą następującego kodu, ale nadal otrzymuję wszystkie dane wyjściowe z ładowania sieci. Jak poprawnie wyłączyć wyjście?Ustawienie GLOG_minloglevel = 1, aby zapobiec wyjściu z powłoki z Caffe

os.environ["GLOG_minloglevel"] = "1" 
net = caffe.Net(model_file, pretrained, caffe.TEST) 
os.environ["GLOG_minloglevel"] = "0" 

Odpowiedz

37

Aby wyciszyć poziom wyjściowy trzeba wzrost poziomu logowania do co najmniej 2

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

Poziomy są

0 - debug
1 - info (jeszcze dużo wyjść)
2 - ostrzeżenia
3 - błędy


Aktualizacja:
Ponieważ ta flaga jest globalny do caffe należy ustawić przed do importowania z caffe pakietu (jak podkreślił jbum). Po ustawieniu flagi i zaimportowaniu caffe zachowanie narzędzia GLOG nie może zostać zmienione.

24

udało mi się dostać Shai's solution do pracy, ale tylko przez wykonanie tej linii w Pythonie przed wywołującego

import caffe 
Powiązane problemy