  
  import cv2
  import numpy as np
  
  import tensorflow as tf
  from tensorflow.python import errors_impl as tf_errors  # pylint:disable=no-name-in-module
  
  from lib.alignments import Alignments
  from lib.faces_detect import DetectedFace
  from lib.training_data import TrainingDataGenerator
  from lib.utils import FaceswapError, get_folder, get_image_paths
  from plugins.train._config import Config
 
</s> remove from lib.model.masks import get_available_masks
 </s> add  </s> remove from lib.model import masks
 </s> add  </s> remove from lib.model.masks import get_available_masks, get_default_mask
 </s> add  </s> remove from lib.model.masks import get_available_masks
 </s> add 
  
  import cv2
  import numpy as np
  



  from lib.alignments import Alignments
  from lib.faces_detect import DetectedFace
  from lib.training_data import TrainingDataGenerator
  from lib.utils import FaceswapError, get_folder, get_image_paths
  from plugins.train._config import Config
keep keep keep keep replace replace replace keep keep keep keep keep
 Smart Mask Exposure for Extraction & Training (#831)

Smart Masks - Initial Commit 
plugins/train/trainer/_base.py