Преобразовать файл прототипа caffe в тензор - PullRequest
Новая
       25

Преобразовать файл прототипа caffe в тензор

0 голосов
/ 11 июня 2019

У меня есть файл prototxt, определяющий модель кафе, образец которого вставлен ниже.Я хотел бы сценарий, который может анализировать этот файл и производить код в tenorsflow python, который устанавливает эти слои и соединяет их, как в этом файле.

Я опробовал пакет caffe-tenorflow в этом посте Eliot Andres: https://ndres.me/post/convert-caffe-to-tensorflow/

У меня нет моделей, фактически обученных кофе, я надеюсь обучить их непосредственно в тензорном потоке.Поэтому я попытался запустить эту команду:

python caffe-tensorflow/convert.py file.prototxt --code-output-path ./output.py

, которая вызвала ошибку: 

Traceback (most recent call last):
  File "caffe-tensorflow/convert.py", line 130, in <module>
    main()
  File "caffe-tensorflow/convert.py", line 126, in main
    args.standalone_output_path, args.phase)
  File "caffe-tensorflow/convert.py", line 32, in convert
    transformer = TensorFlowTransformer(def_path, caffemodel_path, phase=phase)
  File "/home/home1/user/tensorflow_translation/caffe-tensorflow/kaffe/tensorflow/transformer.py", line 221, in __init__
    self.load(def_path, data_path, phase)
  File "/home/home1/user/tensorflow_translation/caffe-tensorflow/kaffe/tensorflow/transformer.py", line 227, in load
    graph = GraphBuilder(def_path, phase).build()
  File "/home/home1/user/tensorflow_translation/caffe-tensorflow/kaffe/graph.py", line 209, in build
    layers = self.filter_layers(layers)
  File "/home/home1/user/tensorflow_translation/caffe-tensorflow/kaffe/graph.py", line 168, in filter_layers
    assert layer.name not in filtered_layer_names
AssertionError```


Snippet from file.prototxt:

```python
name: "OSVOS"
layer {
  name: "data"
  type: "Python"
  top: "data"
  top: "label"
  top: "mbcat"
  top: "of"
  top: "dt"
  top: "label_dt"
  include {
    phase: TRAIN
  }
  python_param {
    module: "needle_input_layer_batch_celoss_dt_side"
    layer: "OSVOSDataLayer"
    param_str: "{\'data_root_dir\': \'/usr/xtmp/user/DATA/DAVIS/DAVIS-2017-trainval-480p/\', \'seed\': 1337,\'pair_list\':\'./solvers/train_6_dt.txt\',\'batchSize\':8}"
  }
}

Может кто-нибудь, пожалуйста, помогите мне, как использовать пакет caffe-tenorflowflowправильно, или указать мне на любые другие пакеты, которые могут сделать это?Большое спасибо.

...