Set the input channel order for e.g. RGB to BGR conversion as needed for the reference ImageNet model. Parameters ---------- in_ : which input to assign this channel order order : the order to transpose the dimensions
(self, in_, order)
| 185 | return decaf_in |
| 186 | |
| 187 | def set_transpose(self, in_, order): |
| 188 | """ |
| 189 | Set the input channel order for e.g. RGB to BGR conversion |
| 190 | as needed for the reference ImageNet model. |
| 191 | |
| 192 | Parameters |
| 193 | ---------- |
| 194 | in_ : which input to assign this channel order |
| 195 | order : the order to transpose the dimensions |
| 196 | """ |
| 197 | self.__check_input(in_) |
| 198 | if len(order) != len(self.inputs[in_]) - 1: |
| 199 | raise Exception('Transpose order needs to have the same number of ' |
| 200 | 'dimensions as the input.') |
| 201 | self.transpose[in_] = order |
| 202 | |
| 203 | def set_channel_swap(self, in_, order): |
| 204 | """ |