diff --git a/mrcnn/model.py b/mrcnn/model.py index 7e68389ebdef733d0cdb5e043f8db6fde728f695..f58e2a68b2b79308945d94ff81336107b0e74df4 100644 --- a/mrcnn/model.py +++ b/mrcnn/model.py @@ -2274,10 +2274,6 @@ class MaskRCNN(): self.log_dir = os.path.join(self.model_dir, "{}{:%Y%m%dT%H%M}".format( self.config.NAME.lower(), now)) - # Create log_dir if not exists - if not os.path.exists(self.log_dir): - os.makedirs(self.log_dir) - # Path to save after each epoch. Include placeholders that get filled by Keras. self.checkpoint_path = os.path.join(self.log_dir, "mask_rcnn_{}_*epoch*.h5".format( self.config.NAME.lower())) @@ -2342,6 +2338,10 @@ class MaskRCNN(): val_generator = data_generator(val_dataset, self.config, shuffle=True, batch_size=self.config.BATCH_SIZE) + # Create log_dir if it does not exist + if not os.path.exists(self.log_dir): + os.makedirs(self.log_dir) + # Callbacks callbacks = [ keras.callbacks.TensorBoard(log_dir=self.log_dir,