plugins package

The plugins package holds Extraction, Training and Conversion plugins for Faceswap.

• convert package
  • mask package
    • mask.mask_blend module
      • Mask
        • Mask.run()
  • writer package
    • writer._base module
      • Output
        • Output.cache_frame()
        • Output.close()
        • Output.is_stream
        • Output.output_filename()
        • Output.pre_encode()
        • Output.write()
      • get_config()
    • writer.ffmpeg module
      • Writer
        • Writer.close()
        • Writer.write()
    • writer.gif module
      • Writer
        • Writer.close()
        • Writer.write()
    • writer.opencv module
      • Writer
        • Writer.close()
        • Writer.pre_encode()
        • Writer.write()
    • writer.patch module
      • Writer
        • Writer.close()
        • Writer.pre_encode()
        • Writer.write()
    • writer.pillow module
      • Writer
        • Writer.close()
        • Writer.pre_encode()
        • Writer.write()
• extract package
  • extract_media module
    • ExtractMedia
      • ExtractMedia.add_detected_faces()
      • ExtractMedia.add_frame_metadata()
      • ExtractMedia.add_sub_folders()
      • ExtractMedia.detected_faces
      • ExtractMedia.filename
      • ExtractMedia.frame_metadata
      • ExtractMedia.get_image_copy()
      • ExtractMedia.image
      • ExtractMedia.image_shape
      • ExtractMedia.image_size
      • ExtractMedia.is_aligned
      • ExtractMedia.remove_image()
      • ExtractMedia.set_image()
      • ExtractMedia.sub_folders
  • pipeline module
    • Extractor
      • Extractor.phase
      • Extractor.aligner
      • Extractor.detected_faces()
      • Extractor.final_pass
      • Extractor.import_data()
      • Extractor.input_queue
      • Extractor.launch()
      • Extractor.passes
      • Extractor.phase_text
      • Extractor.recognition
      • Extractor.reset_phase_index()
      • Extractor.set_batchsize()
  • _base module
    • Extractor
      • Extractor.name
      • Extractor.input_size
      • Extractor.color_format
      • Extractor.vram
      • Extractor.vram_warnings
      • Extractor.vram_per_batch
      • Extractor.batchsize
      • Extractor.check_and_raise_error()
      • Extractor.config
      • Extractor.finalize()
      • Extractor.get_batch()
      • Extractor.init_model()
      • Extractor.initialize()
      • Extractor.join()
      • Extractor.model
      • Extractor.model_path
      • Extractor.on_completion()
      • Extractor.predict()
      • Extractor.process_input()
      • Extractor.process_output()
      • Extractor.queue_size
      • Extractor.rollover_collector()
      • Extractor.start()
    • ExtractorBatch
      • ExtractorBatch.data
      • ExtractorBatch.detected_faces
      • ExtractorBatch.feed
      • ExtractorBatch.filename
      • ExtractorBatch.image
      • ExtractorBatch.prediction
  • align plugins package
    • align._base.aligner module
      • Aligner
        • Aligner.faces_to_feed()
        • Aligner.finalize()
        • Aligner.get_batch()
        • Aligner.initialize()
        • Aligner.on_completion()
        • Aligner.set_normalize_method()
      • AlignerBatch
        • AlignerBatch.batch_id
        • AlignerBatch.detected_faces
        • AlignerBatch.landmarks
        • AlignerBatch.refeeds
        • AlignerBatch.second_pass
        • AlignerBatch.second_pass_masks
    • align._base.processing module
      • AlignedFilter
        • AlignedFilter.filtered_mask()
        • AlignedFilter.output_counts()
      • ReAlign
        • ReAlign.active
        • ReAlign.add_batch()
        • ReAlign.do_filter
        • ReAlign.do_refeeds
        • ReAlign.get_batch()
        • ReAlign.items_queued
        • ReAlign.items_tracked
        • ReAlign.process_batch()
        • ReAlign.process_output()
        • ReAlign.set_input_size_and_centering()
        • ReAlign.track_batch()
        • ReAlign.untrack_batch()
    • align.cv2_dnn module
      • Align
        • Align.align_image()
        • Align.faces_to_feed()
        • Align.get_pts_from_predict()
        • Align.get_square_box()
        • Align.init_model()
        • Align.move_box()
        • Align.pad_image()
        • Align.predict()
        • Align.process_input()
        • Align.process_output()
    • align.fan module
      • Align
        • Align.crop()
        • Align.faces_to_feed()
        • Align.get_center_scale()
        • Align.get_pts_from_predict()
        • Align.init_model()
        • Align.predict()
        • Align.process_input()
        • Align.process_output()
        • Align.transform()
  • detect plugins package
    • detect._base module
      • Detector
        • Detector.finalize()
        • Detector.get_batch()
      • DetectorBatch
        • DetectorBatch.detected_faces
        • DetectorBatch.initial_feed
        • DetectorBatch.pad
        • DetectorBatch.rotation_matrix
        • DetectorBatch.scale
    • detect.mtcnn module
      • Detect
        • Detect.init_model()
        • Detect.predict()
        • Detect.process_input()
        • Detect.process_output()
      • MTCNN
        • MTCNN.detect_faces()
      • ONet
        • ONet.model_definition()
      • PNet
        • PNet.model_definition()
      • RNet
        • RNet.model_definition()
      • nms()
      • rect2square()
  • mask plugins package
    • mask._base module
      • Masker
        • Masker.finalize()
        • Masker.get_batch()
      • MaskerBatch
        • MaskerBatch.detected_faces
        • MaskerBatch.feed_faces
        • MaskerBatch.roi_masks
    • mask.bisenet_fp module
      • AttentionRefinementModule
      • BiSeNet
      • BiSeNetOutput
      • ContextPath
      • ConvBn
      • FeatureFusionModule
      • Mask
        • Mask.init_model()
        • Mask.predict()
        • Mask.process_input()
        • Mask.process_output()
      • ResNet18
  • recognition plugins package
    • recognition._base module
      • Identity
        • Identity.add_identity_filters()
        • Identity.finalize()
        • Identity.get_batch()
      • IdentityFilter
        • IdentityFilter.add_filters()
        • IdentityFilter.output_counts()
      • RecogBatch
        • RecogBatch.detected_faces
        • RecogBatch.feed_faces
    • recognition.vgg_face2 module
      • Cluster
      • Recognition
        • Recognition.init_model()
        • Recognition.predict()
        • Recognition.process_input()
        • Recognition.process_output()
• plugin_loader module
  • PluginLoader
    • PluginLoader.get_aligner()
    • PluginLoader.get_available_convert_plugins()
    • PluginLoader.get_available_extractors()
    • PluginLoader.get_available_models()
    • PluginLoader.get_converter()
    • PluginLoader.get_default_model()
    • PluginLoader.get_detector()
    • PluginLoader.get_masker()
    • PluginLoader.get_model()
    • PluginLoader.get_recognition()
    • PluginLoader.get_trainer()
• train package
  • model package
    • model._base.model module
      • ModelBase
        • ModelBase.input_shape
        • ModelBase.trainer
        • ModelBase.add_history()
        • ModelBase.build()
        • ModelBase.build_model()
        • ModelBase.command_line_arguments
        • ModelBase.config
        • ModelBase.coverage_ratio
        • ModelBase.input_shapes
        • ModelBase.io
        • ModelBase.iterations
        • ModelBase.model
        • ModelBase.model_name
        • ModelBase.name
        • ModelBase.output_shapes
        • ModelBase.state
      • State
        • State.add_mixed_precision_layers()
        • State.add_session_batchsize()
        • State.add_session_loss_names()
        • State.current_session
        • State.filename
        • State.increment_iterations()
        • State.iterations
        • State.loss_names
        • State.lowest_avg_loss
        • State.mixed_precision_layers
        • State.model_needs_rebuild
        • State.save()
        • State.session_id
        • State.sessions
        • State.update_session_config()
    • model._base.settings module
      • Loss
        • Loss.configure()
        • Loss.functions
        • Loss.names
      • LossClass
        • LossClass.function()
        • LossClass.init
        • LossClass.kwargs
      • Optimizer
        • Optimizer.optimizer
      • Settings
        • Settings.check_model_precision()
        • Settings.get_mixed_precision_layers()
        • Settings.loss_scale_optimizer()
        • Settings.strategy_scope()
        • Settings.use_mixed_precision
    • model._base.io module
      • IO
        • IO.filename
        • IO.history
        • IO.load()
        • IO.model_dir
        • IO.model_exists
        • IO.multiple_models_in_folder
        • IO.save()
        • IO.snapshot()
      • Weights
        • Weights.freeze()
        • Weights.load()
      • get_all_sub_models()
    • model.original module
      • Model
        • Model.build_model()
        • Model.decoder()
        • Model.encoder()
  • trainer package
    • trainer._base module
      • TrainerBase
        • TrainerBase.clear_tensorboard()
        • TrainerBase.exit_early
        • TrainerBase.toggle_mask()
        • TrainerBase.train_one_step()