(self)
| 30 | } |
| 31 | |
| 32 | def process(self): |
| 33 | for idx, layer in enumerate(self.net.layers): |
| 34 | param = layer.blobs |
| 35 | name = self.layer_names[idx] |
| 36 | if layer.type in self.processors: |
| 37 | logger.info("Processing layer {} of type {}".format( |
| 38 | name, layer.type)) |
| 39 | dic = self.processors[layer.type](idx, name, param) |
| 40 | self.param_dict.update(dic) |
| 41 | elif len(layer.blobs) != 0: |
| 42 | logger.warn( |
| 43 | "{} layer contains parameters but is not supported!".format(layer.type)) |
| 44 | return self.param_dict |
| 45 | |
| 46 | def proc_conv(self, idx, name, param): |
| 47 | assert len(param) <= 2 |
no test coverage detected