(caffe_net)
| 181 | |
| 182 | |
| 183 | def _GetInputDims(caffe_net): |
| 184 | input_dims = [] |
| 185 | if caffe_net.input_dim: |
| 186 | input_dims = caffe_net.input_dim |
| 187 | elif caffe_net.input_shape: |
| 188 | input_dims = caffe_net.input_shape[0].dim |
| 189 | elif caffe_net.layer[0].input_param.shape: |
| 190 | # getting input dimension from first layer |
| 191 | input_dims = caffe_net.layer[0].input_param.shape[0].dim |
| 192 | return input_dims |
| 193 | |
| 194 | |
| 195 | class TranslatorRegistry: |
no outgoing calls
no test coverage detected
searching dependent graphs…