Helper functionn to create encoder of the model
(image_height, image_width)
| 325 | return fusion_layer |
| 326 | |
| 327 | def get_encoder(image_height, image_width): |
| 328 | """ |
| 329 | Helper functionn to create encoder of the model |
| 330 | """ |
| 331 | input_layer = tf.keras.layers.Input( |
| 332 | shape=(image_height, image_width, 3), name="input_layer" |
| 333 | ) |
| 334 | |
| 335 | # learning module |
| 336 | learning_layer1, learning_layer2, learning_layer3 = learning_module(input_layer=input_layer) |
| 337 | |
| 338 | # feature extractor module |
| 339 | fe_layer = feature_extractor(learning_layer3) |
| 340 | |
| 341 | # feature fusion module |
| 342 | fusion_layer = fusion_module(learning_layer=learning_layer3, fe_layer=fe_layer) |
| 343 | |
| 344 | # build model |
| 345 | model = tf.keras.Model(inputs=input_layer, outputs=fusion_layer, name="pcb-encoder") |
| 346 | return model, learning_layer1, learning_layer2 |
| 347 | |
| 348 | def get_decoder(encoder, learning_layer1, learning_layer2, num_classes): |
| 349 |
no test coverage detected