2015-06-23 3 views
6

Я использовал код питона рисовать Net, определенный в файле прототекст, как:Caffe: Рисунок CNN Net

python draw_net.py test.protxt test.png 

Он не в состоянии сделать. Он не показывает никаких ошибок, но результат test.png - белый пустой файл изображения. Может ли кто-нибудь помочь мне в его исправлении? Это действительно поможет быстро разработать новые сети.

+1

, если вы используете интерфейс NVIDIA [DIGITS] (https://github.com/NVIDIA/DIGITS) для доступа к caffe, вы сможете получить этот рисунок из пользовательского интерфейса. – Shai

+1

Кстати, насколько велика ваша сеть? можете ли вы привлечь меньшие сети? – Shai

+0

Я могу нарисовать сиамскую сеть (указанную в библиотеке), но не могу рисовать сети для классификации Imagenet (данный в библиотеке тоже) – thetna

ответ

2

У меня была такая же проблема. Основываясь на this thread, мне удалось решить эту проблему, используя старый синтаксис Proto, как было предложено. Например, я должен был это сделать:

Переименовать слои из layers в layer. Весь слой типа переименования по Caffe документации (или на примере прото-файлы) - т.е. слоя type: CONVOLUTION к type: "Convolution" и т.д. Запасной новому синтаксис:

blobs_lr: 1   
blobs_lr: 1   
weight_decay: 1 
weight_decay: 0 

для

param { 
    name: "conv1_w" 
    lr_mult: 1    
    decay_mult: 1 
} 
param { 
    name: "conv1_b" 
    lr_mult: 2    
    decay_mult: 0 
} 

Теперь разбора и нового волочения работает только хорошо. См. Примеры файлов .prototxt в пакете caffe для улучшения интуиции, как выглядит рабочий синтаксис прото.

2

В середине 2014 года Caffe changed their proto definition for extensibility причиняет эту проблему. В результате этого изменения все прото-файлы должны быть обновлены до более нового определения.

Для этого Caffe предоставляет следующие инструменты в distribute/bin/ или .build_release/tools каталоге:

  1. upgrade_net_proto_binary.bin
  2. upgrade_net_proto_text.bin

Вот простой пример того, как преобразовать текстовый файл прото в более новый формат:

./upgrade_net_proto_text.bin /path/to/older_proto_file /path/to/newer_ouput_proto_file 
Смежные вопросы