added anchor parameter changeable definitions to the RetinaNet notebo…

…ok, and fixed a bug where n_semantic_classes was not defined when panoptic==False
vanvalenlab · Jul 4, 2019 · 0d8dc57 · 0d8dc57
1 parent 5858374
commit 0d8dc57
Show file tree

Hide file tree

Showing 2 changed files with 157 additions and 120 deletions.
diff --git a/deepcell/training.py b/deepcell/training.py
@@ -539,6 +539,8 @@ def train_model_retinanet(model,
     if panoptic:
         n_semantic_classes = [layer.output_shape[channel_axis]
                               for layer in model.layers if 'semantic' in layer.name]
+    else:
+        n_semantic_classes = []
 
     # the data, shuffled and split between train and test sets
     print('X_train shape:', train_dict['X'].shape)