Установка GLOG_minloglevel = 1 для предотвращения вывода в оболочке из Caffe
Я использую Caffe, который печатает много вывода в оболочку при загрузке нейронной сети.
Я хотел бы подавить этот вывод, который, возможно, можно сделать, установив GLOG_minloglevel=1
при запуске Python script. Я пробовал сделать это, используя следующий код, но я все равно получаю все выходные данные от загрузки сети. Как правильно подавить вывод?
os.environ["GLOG_minloglevel"] = "1"
net = caffe.Net(model_file, pretrained, caffe.TEST)
os.environ["GLOG_minloglevel"] = "0"
Ответы
Ответ 1
Чтобы подавить выходной уровень, вам нужно увеличить уровень log до 2
os.environ['GLOG_minloglevel'] = '2'
Уровни
0 - debug
1 - информация (все еще много выходов)
2 - предупреждения
3 - ошибки
Update:
Поскольку этот флаг является глобальным для caffe
, он должен быть установлен до импорта пакета caffe
(как указано jbum). Когда флаг установлен и caffe
импортирован, поведение инструмента GLOG не может быть изменено.
Ответ 2
Я смог получить решение Shai для работы, но только выполнив эту строку на Python перед вызовом
import caffe