(encoder, num_classes)
| 391 | return model |
| 392 | |
| 393 | def get_classification(encoder, num_classes): |
| 394 | inputs = encoder.inputs |
| 395 | T1 = encoder.output |
| 396 | gap = tf.keras.layers.GlobalAveragePooling2D()(T1) |
| 397 | d1 = tf.keras.layers.Dense(128, activation="relu")(gap) |
| 398 | d2 = tf.keras.layers.Dense(num_classes, activation="softmax")(d1) |
| 399 | |
| 400 | model = tf.keras.Model(inputs=inputs, outputs=d2, name="pcb-classification") |
| 401 | return model |