Removed obsolete constant N_CLASSES
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
parent
af62d0492c
commit
ab9f3f8a94
|
@ -17,9 +17,6 @@
|
|||
"""
|
||||
Provides functionality to use the SSD Keras implementation.
|
||||
|
||||
Attributes:
|
||||
N_CLASSES: number of known classes (without background)
|
||||
|
||||
Functions:
|
||||
compile_model(...): compiles an SSD model
|
||||
get_loss_func(...): returns the SSD loss function
|
||||
|
@ -45,8 +42,6 @@ from twomartens.masterthesis.ssd_keras.ssd_encoder_decoder import ssd_output_dec
|
|||
K = tf.keras.backend
|
||||
tfe = tf.contrib.eager
|
||||
|
||||
N_CLASSES = 80
|
||||
|
||||
|
||||
def get_model(use_dropout: bool,
|
||||
dropout_model: callable, vanilla_model: callable,
|
||||
|
|
Loading…
Reference in New Issue