  from scripts.train import TrainingProcessor
  from scripts.convert import ConvertImage
  
  if __name__ == "__main__":
      parser = argparse.ArgumentParser()
      subparser = parser.add_subparsers()
      extract = ExtractTrainingData(
          subparser, "extract", "Extract the faces from a pictures.")
      train = TrainingProcessor(
          subparser, "train", "This command trains the model for the two faces A and B.")
 
</s> remove     try:
        arguments.func(arguments)
    except:
        parser.print_help()
 </s> add     arguments.func(arguments) </s> remove import argparse
 </s> add from lib.utils import FullHelpArgumentParser
 </s> remove     def process_image(self, filename):
        # TODO move the model load and the converter creation in a method called on init, but after the arg parsing
        (face_A,face_B) = ('/decoder_A.h5', '/decoder_B.h5') if not self.arguments.swap_model else ('/decoder_B.h5', '/decoder_A.h5')
 </s> add     def process(self):
        # Original model goes with Adjust or Masked converter
        model_name = "Original" # TODO Pass as argument
        conv_name = "Masked" # TODO Pass as argument </s> remove             self.process_image(filename)
 </s> add             yield filename
    parser = FullHelpArgumentParser()
  from scripts.train import TrainingProcessor
  from scripts.convert import ConvertImage
  
  if __name__ == "__main__":
    parser = FullHelpArgumentParser()
      subparser = parser.add_subparsers()
      extract = ExtractTrainingData(
          subparser, "extract", "Extract the faces from a pictures.")
      train = TrainingProcessor(
          subparser, "train", "This command trains the model for the two faces A and B.")
keep keep keep keep replace keep keep keep keep keep
 Adding models as plugins + Face filtering (#53) + #39 + #43 + #44 + #49 (#61)

* Making Models as plugins

* Do not reload model on each image #39 + Adding FaceFilter #53

* Adding @lukaville PR for #43 and #44 (possibly)

* Training done in a separate thread

* Better log for plugin load

* Adding a prefetch to train.py #49
(Note that we prefetch 2 batches of images, due to the queue behavior)
+ More compact logging with verbose info included

* correction of DirectoryProcessor signature

* adding missing import

* Convert with parallel preprocessing of files

* Added coverage var for trainer

Added a var with comment. Feel free to add it as argument

* corrections

* Modifying preview and normalization of image + correction

* Cleanup 
faceswap.py