2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Removed obsolete constant N_CLASSES

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-11 16:06:05 +02:00
parent af62d0492c
commit ab9f3f8a94
1 changed files with 0 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/twomartens/masterthesis/ssd.py
View File

@ -17,9 +17,6 @@
""" """
Provides functionality to use the SSD Keras implementation. Provides functionality to use the SSD Keras implementation.
Attributes:
N_CLASSES: number of known classes (without background)
Functions: Functions:
compile_model(...): compiles an SSD model compile_model(...): compiles an SSD model
get_loss_func(...): returns the SSD loss function 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 K = tf.keras.backend
tfe = tf.contrib.eager tfe = tf.contrib.eager
N_CLASSES = 80
def get_model(use_dropout: bool, def get_model(use_dropout: bool,
dropout_model: callable, vanilla_model: callable, dropout_model: callable, vanilla_model: callable,