build encoder, decoder and final model
(self)
| 26 | self.num_classes = opt["train"]["num_classes"] + 1 |
| 27 | |
| 28 | def build(self): |
| 29 | """ |
| 30 | build encoder, decoder and final model |
| 31 | """ |
| 32 | encoder, learning_layer1, learning_layer2 = get_encoder(self.image_height, self.image_width) |
| 33 | model = get_decoder(encoder, learning_layer1, learning_layer2, self.num_classes) |
| 34 | print(model.summary()) |
| 35 | return model |
| 36 | |
| 37 | class PCBClassNet: |
| 38 | """ |
no test coverage detected