Method
TranslateLayer
(cls, layer, pretrained_blobs, is_test, **kwargs)
Source from the content-addressed store, hash-verified
| 207 | |
| 208 | @classmethod |
| 209 | def TranslateLayer(cls, layer, pretrained_blobs, is_test, **kwargs): |
| 210 | try: |
| 211 | caffe_ops, params = cls.registry_[layer.type]( |
| 212 | layer, pretrained_blobs, is_test, **kwargs) |
| 213 | except KeyError as e: |
| 214 | raise KeyError('No translator registered for layer: %s yet.' % |
| 215 | str(layer)) from e |
| 216 | if caffe_ops is None: |
| 217 | caffe_ops = [] |
| 218 | if type(caffe_ops) is not list: |
| 219 | caffe_ops = [caffe_ops] |
| 220 | return caffe_ops, params |
| 221 | |
| 222 | @classmethod |
| 223 | def TranslateModel( |
Tested by
no test coverage detected