DeepFaceLab fork which provides IPython Notebook to use DFL with Google Colab

Overview

DFL-Colab — DeepFaceLab fork for Google Colab

This project provides you IPython Notebook to use DeepFaceLab with Google Colaboratory.
You can create your deepfake video even if you don’t have a powerful GPU on a desktop PC.

Keep in mind that working with a remote server imposes some restrictions.
Because of this, working with DFL-Colab is not so convenient as working with DeepFaceLab directly on your computer.

DFL-Colab uses DeepFaceLab and to understand the workflow, you should first understand the DeepFaceLab workflow.

Official Links

  • DFL-Colab Notebook: Link -- Works with latest DFL version
  • DFL-Colab Notebook 1.0: Link -- Works with DFL 1.0 version
  • DeepFaceLab Repository: Link
  • Youtube channel: Link

Community support

  • Telegram Chat (English/Russian): Link
  • MrDeepFakes Forum: Link
  • MrDeepFakes DFL-Colab Guide: Link
  • MrDeepFakes DeepFaceLab Guide: Link
  • Reddit — r/GifFakes/: Link
  • Reddit — r/SFWdeepfakes/: Link
Comments
  • SAEHD Training Failure

    SAEHD Training Failure

    Model Training (SAEHD and Quick96) fails upon execution.

    1. DLF installation functions
    2. Frame extraction functions
    3. DST and SRC face functions
    4. Xseg training functions
    5. Xseg apply/remove functions
    6. Model training fails
    • Sample error code

    Process Process-11: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last): File "/usr/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap self.run() File "/usr/lib/python3.8/multiprocessing/process.py", line 108, in run self._target(*self._args, **self._kwargs) File "/content/DeepFaceLab/core/joblib/SubprocessGenerator.py", line 54, in process_func gen_data = next (self.generator_func) File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 136, in batch_func raise Exception ("Exception occured in sample %s. Error: %s" % (sample.filename, traceback.format_exc() ) ) Exception: Exception occured in sample /content/workspace/data_src/aligned/00973.jpg. Error: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    Process Process-14: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last): File "/usr/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap self.run() File "/usr/lib/python3.8/multiprocessing/process.py", line 108, in run self._target(*self._args, **self._kwargs) File "/content/DeepFaceLab/core/joblib/SubprocessGenerator.py", line 54, in process_func gen_data = next (self.generator_func) File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 136, in batch_func raise Exception ("Exception occured in sample %s. Error: %s" % (sample.filename, traceback.format_exc() ) ) Exception: Exception occured in sample /content/workspace/data_src/aligned/00402.jpg. Error: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    Process Process-15: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last): File "/usr/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap self.run() File "/usr/lib/python3.8/multiprocessing/process.py", line 108, in run self._target(*self._args, **self._kwargs) File "/content/DeepFaceLab/core/joblib/SubprocessGenerator.py", line 54, in process_func gen_data = next (self.generator_func) File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 136, in batch_func raise Exception ("Exception occured in sample %s. Error: %s" % (sample.filename, traceback.format_exc() ) ) Exception: Exception occured in sample /content/workspace/data_src/aligned/01929.jpg. Error: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    Process Process-12: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last): File "/usr/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap self.run() File "/usr/lib/python3.8/multiprocessing/process.py", line 108, in run self._target(*self._args, **self._kwargs) File "/content/DeepFaceLab/core/joblib/SubprocessGenerator.py", line 54, in process_func gen_data = next (self.generator_func) File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 136, in batch_func raise Exception ("Exception occured in sample %s. Error: %s" % (sample.filename, traceback.format_exc() ) ) Exception: Exception occured in sample /content/workspace/data_src/aligned/00293.jpg. Error: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    Process Process-16: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last): File "/usr/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap self.run() File "/usr/lib/python3.8/multiprocessing/process.py", line 108, in run self._target(*self._args, **self._kwargs) File "/content/DeepFaceLab/core/joblib/SubprocessGenerator.py", line 54, in process_func gen_data = next (self.generator_func) File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 136, in batch_func raise Exception ("Exception occured in sample %s. Error: %s" % (sample.filename, traceback.format_exc() ) ) Exception: Exception occured in sample /content/workspace/data_src/aligned/00412.jpg. Error: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    Process Process-13: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last): File "/usr/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap self.run() File "/usr/lib/python3.8/multiprocessing/process.py", line 108, in run self._target(*self._args, **self._kwargs) File "/content/DeepFaceLab/core/joblib/SubprocessGenerator.py", line 54, in process_func gen_data = next (self.generator_func) File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 136, in batch_func raise Exception ("Exception occured in sample %s. Error: %s" % (sample.filename, traceback.format_exc() ) ) Exception: Exception occured in sample /content/workspace/data_src/aligned/02330.jpg. Error: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    Loading samples: 100% 308/308 [00:00<00:00, 678.14it/s] Process Process-26: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last): File "/usr/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap self.run() File "/usr/lib/python3.8/multiprocessing/process.py", line 108, in run self._target(*self._args, **self._kwargs) File "/content/DeepFaceLab/core/joblib/SubprocessGenerator.py", line 54, in process_func gen_data = next (self.generator_func) File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 136, in batch_func raise Exception ("Exception occured in sample %s. Error: %s" % (sample.filename, traceback.format_exc() ) ) Exception: Exception occured in sample /content/workspace/data_dst/aligned/00014_0.jpg. Error: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    Process Process-27: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last): File "/usr/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap self.run() File "/usr/lib/python3.8/multiprocessing/process.py", line 108, in run self._target(*self._args, **self._kwargs) File "/content/DeepFaceLab/core/joblib/SubprocessGenerator.py", line 54, in process_func gen_data = next (self.generator_func) File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 136, in batch_func raise Exception ("Exception occured in sample %s. Error: %s" % (sample.filename, traceback.format_exc() ) ) Exception: Exception occured in sample /content/workspace/data_dst/aligned/00108_0.jpg. Error: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    Process Process-25: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last): File "/usr/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap self.run() File "/usr/lib/python3.8/multiprocessing/process.py", line 108, in run self._target(*self._args, **self._kwargs) File "/content/DeepFaceLab/core/joblib/SubprocessGenerator.py", line 54, in process_func gen_data = next (self.generator_func) File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 136, in batch_func raise Exception ("Exception occured in sample %s. Error: %s" % (sample.filename, traceback.format_exc() ) ) Exception: Exception occured in sample /content/workspace/data_dst/aligned/00054_0.jpg. Error: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    Process Process-29: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last): File "/usr/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap self.run() File "/usr/lib/python3.8/multiprocessing/process.py", line 108, in run self._target(*self._args, **self._kwargs) File "/content/DeepFaceLab/core/joblib/SubprocessGenerator.py", line 54, in process_func gen_data = next (self.generator_func) File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 136, in batch_func raise Exception ("Exception occured in sample %s. Error: %s" % (sample.filename, traceback.format_exc() ) ) Exception: Exception occured in sample /content/workspace/data_dst/aligned/00276_0.jpg. Error: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    Process Process-30: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last): File "/usr/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap self.run() File "/usr/lib/python3.8/multiprocessing/process.py", line 108, in run self._target(*self._args, **self._kwargs) File "/content/DeepFaceLab/core/joblib/SubprocessGenerator.py", line 54, in process_func gen_data = next (self.generator_func) File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 136, in batch_func raise Exception ("Exception occured in sample %s. Error: %s" % (sample.filename, traceback.format_exc() ) ) Exception: Exception occured in sample /content/workspace/data_dst/aligned/00209_0.jpg. Error: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    Process Process-28: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last): File "/usr/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap self.run() File "/usr/lib/python3.8/multiprocessing/process.py", line 108, in run self._target(*self._args, **self._kwargs) File "/content/DeepFaceLab/core/joblib/SubprocessGenerator.py", line 54, in process_func gen_data = next (self.generator_func) File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 136, in batch_func raise Exception ("Exception occured in sample %s. Error: %s" % (sample.filename, traceback.format_exc() ) ) Exception: Exception occured in sample /content/workspace/data_dst/aligned/00189_0.jpg. Error: Traceback (most recent call last): File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 134, in batch_func x, = SampleProcessor.process ([sample], self.sample_process_options, self.output_sample_types, self.debug, ct_sample=ct_sample) File "/content/DeepFaceLab/samplelib/SampleProcessor.py", line 106, in process warp_params = imagelib.gen_warp_params(resolution, File "/content/DeepFaceLab/core/imagelib/warp.py", line 146, in gen_warp_params random_transform_mat = cv2.getRotationMatrix2D((w // 2, w // 2), rotation, scale) TypeError: Can't parse 'center'. Sequence item with index 0 has a wrong type

    opened by fulanito3030 6
  • pretrain faceset download error

    pretrain faceset download error

    I am trying to run the first step on the DFL colab latest notebook but get this error on the last part of it. DeepFaceLab uses faceset.pak which could simplify things, not needing to unzip etc.

    I'll try to figure it out locally and create a pull request

    [/content/pretrain_CelebA.zip]
      End-of-central-directory signature not found.  Either this file is not
      a zipfile, or it constitutes one disk of a multi-part archive.  In the
      latter case the central directory and zipfile comment will be found on
      the last disk(s) of this archive.
    unzip:  cannot find zipfile directory in one of /content/pretrain_CelebA.zip or
            /content/pretrain_CelebA.zip.zip, and cannot find /content/pretrain_CelebA.zip.ZIP, period.
    
    opened by camjac251 6
  • TypeError: object of type 'PosixPath' has no len()

    TypeError: object of type 'PosixPath' has no len()

    OS: Debian Linux GPU: Nvidia T4 CUDA: 10.0.130-1 Python: 3.6 Command line: python3.6 DeepFaceLab/main.py merge --input-dir workspace/data_dst --output-dir workspace/data_dst/merged --output-mask-dir workspace/data_dst/merged_mask --aligned-dir workspace/data_dst/aligned --model-dir workspace/model --model "SAEHD"

    I've been training a model for the past 48 hours and I have already merged the model and exported it to a video twice with no problems. I just tried to merge the model again and now it gets to frame 95 and crashes. See the error log below this paragraph. Is my model useless now? I've been training it on a Google Compute VM so if the model is useless then I've wasted $10 on the Google Compute VM so far.

    Merging: 3% 95/3623 [00:05<03:27, 17.00it/s]Exception while process data [/content/workspace/data_dst/00096.jpg]: Traceback (most recent call last): File "/content/DeepFaceLab/mainscripts/Merger.py", line 151, in process_data final_img = MergeMasked (self.predictor_func, self.predictor_input_shape, cfg, frame_info) File "/content/DeepFaceLab/merger/MergeMasked.py", line 327, in MergeMasked out_img, out_img_merging_mask = MergeMaskedFace (predictor_func, predictor_input_shape, cfg, frame_info, img_bgr_uint8, img_bgr, img_landmarks) File "/content/DeepFaceLab/merger/MergeMasked.py", line 13, in MergeMaskedFace img_face_mask_a = LandmarksProcessor.get_image_hull_mask (img_bgr.shape, img_face_landmarks) File "/content/DeepFaceLab/facelib/LandmarksProcessor.py", line 355, in get_image_hull_mask lmrks = expand_eyebrows(image_landmarks, eyebrows_expand_mod) File "/content/DeepFaceLab/facelib/LandmarksProcessor.py", line 324, in expand_eyebrows if len(lmrks) != 68: TypeError: object of type 'PosixPath' has no len()

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last): File "/content/DeepFaceLab/core/joblib/SubprocessorBase.py", line 71, in _subprocess_run result = self.process_data (data) File "/content/DeepFaceLab/mainscripts/Merger.py", line 157, in process_data raise Exception( f'Error while merging file [{filepath}]: {e_str}' ) Exception: Error while merging file [/content/workspace/data_dst/00096.jpg]: Traceback (most recent call last): File "/content/DeepFaceLab/mainscripts/Merger.py", line 151, in process_data final_img = MergeMasked (self.predictor_func, self.predictor_input_shape, cfg, frame_info) File "/content/DeepFaceLab/merger/MergeMasked.py", line 327, in MergeMasked out_img, out_img_merging_mask = MergeMaskedFace (predictor_func, predictor_input_shape, cfg, frame_info, img_bgr_uint8, img_bgr, img_landmarks) File "/content/DeepFaceLab/merger/MergeMasked.py", line 13, in MergeMaskedFace img_face_mask_a = LandmarksProcessor.get_image_hull_mask (img_bgr.shape, img_face_landmarks) File "/content/DeepFaceLab/facelib/LandmarksProcessor.py", line 355, in get_image_hull_mask lmrks = expand_eyebrows(image_landmarks, eyebrows_expand_mod) File "/content/DeepFaceLab/facelib/LandmarksProcessor.py", line 324, in expand_eyebrows if len(lmrks) != 68: TypeError: object of type 'PosixPath' has no len()

    /usr/lib/python3.6/multiprocessing/semaphore_tracker.py:143: UserWarning: semaphore_tracker: There appear to be 1 leaked semaphores to clean up at shutdown len(cache)) Exception while process data [/content/workspace/data_dst/00097.jpg]: Traceback (most recent call last): File "/content/DeepFaceLab/mainscripts/Merger.py", line 151, in process_data final_img = MergeMasked (self.predictor_func, self.predictor_input_shape, cfg, frame_info) File "/content/DeepFaceLab/merger/MergeMasked.py", line 327, in MergeMasked out_img, out_img_merging_mask = MergeMaskedFace (predictor_func, predictor_input_shape, cfg, frame_info, img_bgr_uint8, img_bgr, img_landmarks) File "/content/DeepFaceLab/merger/MergeMasked.py", line 13, in MergeMaskedFace img_face_mask_a = LandmarksProcessor.get_image_hull_mask (img_bgr.shape, img_face_landmarks) File "/content/DeepFaceLab/facelib/LandmarksProcessor.py", line 355, in get_image_hull_mask lmrks = expand_eyebrows(image_landmarks, eyebrows_expand_mod) File "/content/DeepFaceLab/facelib/LandmarksProcessor.py", line 324, in expand_eyebrows if len(lmrks) != 68: TypeError: object of type 'PosixPath' has no len()

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last): File "/content/DeepFaceLab/core/joblib/SubprocessorBase.py", line 71, in _subprocess_run result = self.process_data (data) File "/content/DeepFaceLab/mainscripts/Merger.py", line 157, in process_data raise Exception( f'Error while merging file [{filepath}]: {e_str}' ) Exception: Error while merging file [/content/workspace/data_dst/00097.jpg]: Traceback (most recent call last): File "/content/DeepFaceLab/mainscripts/Merger.py", line 151, in process_data final_img = MergeMasked (self.predictor_func, self.predictor_input_shape, cfg, frame_info) File "/content/DeepFaceLab/merger/MergeMasked.py", line 327, in MergeMasked out_img, out_img_merging_mask = MergeMaskedFace (predictor_func, predictor_input_shape, cfg, frame_info, img_bgr_uint8, img_bgr, img_landmarks) File "/content/DeepFaceLab/merger/MergeMasked.py", line 13, in MergeMaskedFace img_face_mask_a = LandmarksProcessor.get_image_hull_mask (img_bgr.shape, img_face_landmarks) File "/content/DeepFaceLab/facelib/LandmarksProcessor.py", line 355, in get_image_hull_mask lmrks = expand_eyebrows(image_landmarks, eyebrows_expand_mod) File "/content/DeepFaceLab/facelib/LandmarksProcessor.py", line 324, in expand_eyebrows if len(lmrks) != 68: TypeError: object of type 'PosixPath' has no len()

    /usr/lib/python3.6/multiprocessing/semaphore_tracker.py:143: UserWarning: semaphore_tracker: There appear to be 1 leaked semaphores to clean up at shutdown len(cache))

    opened by l0ophole 6
  • Google Colab Auto Deletes

    Google Colab Auto Deletes "DeepFaceLab" Folder

    Merry Christmas, lovely fellow nerds.

    Is it just me, or is Google Colab on purpose messing with us?

    I'm paid Premium++ and it seems they even mess with their paying customers....

    It happened a few times in a row today...That the "DeepFaceLab" folder got deleted instantly after installation and my script errors out "No folder named 'content/DeepFaceLab'" etc.

    And yesterday a few times it "stopped" during the Extraction of frames from a video. With no explanation whatsoever. The random stops during Frame Extraction were not a RAM or Harddisk space problem. The full video is cut into 10-minute parts. I don't even use any GPU for extracting frames so I don't understand why they seem to sabotage my work.

    Let's assume they delete "DeepFaceLab" on purpose, for a moment: They don't know if I'm using DeepFaceLab for company productions or hobby projects. How could they defend deleting "DeepFaceLab" the second after I installed it?

    • What is your experience friends and what are your thoughts? 🤔

    @chervonij

    opened by qkum 5
  • Error: module 'tensorflow' has no attribute 'ConfigProto'

    Error: module 'tensorflow' has no attribute 'ConfigProto'

    Have started getting error:

    [CPU] : CPU [0] : Tesla V100-SXM2-16GB

    [0] Which GPU indexes to choose? : 0

    Error: module 'tensorflow' has no attribute 'ConfigProto' Traceback (most recent call last): File "/content/DeepFaceLab/mainscripts/Trainer.py", line 55, in trainerThread debug=debug, File "/content/DeepFaceLab/models/ModelBase.py", line 151, in init nn.initialize(self.device_config) File "/content/DeepFaceLab/core/leras/nn.py", line 101, in initialize config = tf.ConfigProto() AttributeError: module 'tensorflow' has no attribute 'ConfigProto'

    when using %cd ./DeepFaceLab !git checkout d1a5639e9083508207962f9bd1392af145ad7c88

    using that commit used to work fine a few months ago.

    opened by tania010 5
  • Colab disconnected in just 40 minutes

    Colab disconnected in just 40 minutes

    Hello, as I understand it, Colab must let me work for 12 hours, so DFL_Colab makes a backup every 11 hours. Well, only the first day I used Colab allowed me to work 12 hours. Now it goes offline due to Colab usage limits in under 45 minutes.

    The main problem is that when Colab disconnects, the workspace folder disappears and I cannot find a way to make a copy to drive of what has already been worked.

    Somebody help me please? Sorry for my English but I am Spanish.

    opened by reimond16 5
  • ModuleNotFoundError: No module named 'tensorflow.compat.v2

    ModuleNotFoundError: No module named 'tensorflow.compat.v2

    im facing issue with DFL_Colab_1-0.ipynb (because im using amd in my physical pc so i choose DFL_Colab_1-0.ipynb in colab)

    Using TensorFlow backend. Error: No module named 'tensorflow.compat.v2' Traceback (most recent call last): File "/content/DeepFaceLab/mainscripts/Trainer.py", line 52, in trainerThread device_args=device_args) File "/content/DeepFaceLab/models/ModelBase.py", line 142, in init nnlib.import_all(self.device_config) File "/content/DeepFaceLab/nnlib/nnlib.py", line 1613, in import_all nnlib.import_keras(device_config) File "/content/DeepFaceLab/nnlib/nnlib.py", line 229, in import_keras import keras as keras_ File "/usr/local/lib/python3.7/dist-packages/keras/init.py", line 3, in from . import utils File "/usr/local/lib/python3.7/dist-packages/keras/utils/init.py", line 27, in from .multi_gpu_utils import multi_gpu_model File "/usr/local/lib/python3.7/dist-packages/keras/utils/multi_gpu_utils.py", line 7, in from ..layers.merge import concatenate File "/usr/local/lib/python3.7/dist-packages/keras/layers/init.py", line 4, in from ..engine.base_layer import Layer File "/usr/local/lib/python3.7/dist-packages/keras/engine/init.py", line 3, in from .input_layer import Input File "/usr/local/lib/python3.7/dist-packages/keras/engine/input_layer.py", line 7, in from .base_layer import Layer File "/usr/local/lib/python3.7/dist-packages/keras/engine/base_layer.py", line 12, in from .. import initializers File "/usr/local/lib/python3.7/dist-packages/keras/initializers/init.py", line 17, in import tensorflow.compat.v2 as tf ModuleNotFoundError: No module named 'tensorflow.compat.v2'

    please help

    opened by hargums 4
  • imgaug / albumentations error

    imgaug / albumentations error

    Hi,

    I try to run Install or update DeepFaceLab from Github script but get this error

    ERROR: albumentations 0.1.12 has requirement imgaug<0.2.7,>=0.2.5, but you'll have imgaug 0.2.9 which is incompatible.
    
    opened by Soneji 4
  • ZeroDivisionError: integer division or modulo by zero

    ZeroDivisionError: integer division or modulo by zero

    Running Latest Colab gives following exception while Training Model (Step 4).

    Traceback (most recent call last): File "/content/DeepFaceLab/mainscripts/Trainer.py", line 49, in trainerThread device_args=device_args) File "/content/DeepFaceLab/models/ModelBase.py", line 208, in init self.sample_for_preview = self.generate_next_sample() File "/content/DeepFaceLab/models/ModelBase.py", line 398, in generate_next_sample return [next(generator) for generator in self.generator_list] File "/content/DeepFaceLab/models/ModelBase.py", line 398, in return [next(generator) for generator in self.generator_list] File "/content/DeepFaceLab/samplelib/SampleGeneratorFace.py", line 58, in next generator = self.generators[self.generator_counter % len(self.generators) ] ZeroDivisionError: integer division or modulo by zero Done.

    I had it working before but not sure what has changed.

    opened by somilhyperanna 4
  • Questions about usage

    Questions about usage

    So I started the training last night and closed the browser. After 10 minutes or so, I open the browser again and saw that instance was still running then I closed the browser again. Next day I check the instance only to find out that instance is terminated and no models have been uploaded to the drive.

    What is the correct usage here? Should I keep the browser open for 11 hours? Why model hasn't been uploaded to the drive?

    opened by Greywolf22 4
  • Getting ValueError: operands could not be broadcast together with shapes (144,144,3) (72,72,3) error

    Getting ValueError: operands could not be broadcast together with shapes (144,144,3) (72,72,3) error

    Tried this today and got the error below

    Drive already mounted at /content/drive; to attempt to forcibly remount, call drive.mount("/content/drive", force_remount=True).
    Time to backup: 10 hours
    /content
    Running trainer.
    
    Loading model...
    
    Model first run. Enter model options as default for each run.
    Write preview history? (y/n ?:help skip:n) : y
    Target iteration (skip:unlimited/default) : 
    0
    Batch_size (?:help skip:0) : 
    0
    Feed faces to network sorted by yaw? (y/n ?:help skip:n) : n
    Flip faces randomly? (y/n ?:help skip:y) : n
    Src face scale modifier % ( -30...30, ?:help skip:0) : 
    0
    Resolution ( 64-256 ?:help skip:128) : 144
    Half or Full face? (h/f, ?:help skip:f) : 
    f
    Learn mask? (y/n, ?:help skip:y) : 
    y
    Optimizer mode? ( 1,2,3 ?:help skip:1) : 
    1
    AE architecture (df, liae ?:help skip:df) : 
    df
    AutoEncoder dims (32-1024 ?:help skip:512) : 
    512
    Encoder dims per channel (21-85 ?:help skip:42) : 
    42
    Decoder dims per channel (10-85 ?:help skip:21) : 
    21
    Remove gray border? (y/n, ?:help skip:n) : 
    n
    Use multiscale decoder? (y/n, ?:help skip:n) : y
    Use pixel loss? (y/n, ?:help skip: n ) : 
    n
    Face style power ( 0.0 .. 100.0 ?:help skip:0.00) : 
    0.0
    Background style power ( 0.0 .. 100.0 ?:help skip:0.00) : 
    0.0
    Using TensorFlow backend.
    WARNING:tensorflow:From /usr/local/lib/python3.6/dist-packages/tensorflow/python/framework/op_def_library.py:263: colocate_with (from tensorflow.python.framework.ops) is deprecated and will be removed in a future version.
    Instructions for updating:
    Colocations handled automatically by placer.
    WARNING:tensorflow:From /usr/local/lib/python3.6/dist-packages/tensorflow/python/ops/math_ops.py:3066: to_int32 (from tensorflow.python.ops.math_ops) is deprecated and will be removed in a future version.
    Instructions for updating:
    Use tf.cast instead.
    Loading: 100% 6791/6791 [00:06<00:00, 1049.77it/s]
    Loading: 100% 6791/6791 [00:05<00:00, 1145.45it/s]
    ===== Model summary =====
    == Model name: SAE
    ==
    == Current iteration: 0
    ==
    == Model options:
    == |== write_preview_history : True
    == |== batch_size : 4
    == |== sort_by_yaw : False
    == |== random_flip : False
    == |== resolution : 144
    == |== face_type : f
    == |== learn_mask : True
    == |== optimizer_mode : 1
    == |== archi : df
    == |== ae_dims : 512
    == |== e_ch_dims : 42
    == |== d_ch_dims : 21
    == |== remove_gray_border : False
    == |== multiscale_decoder : True
    == |== pixel_loss : False
    == |== face_style_power : 0.0
    == |== bg_style_power : 0.0
    == Running on:
    == |== [0 : Tesla T4]
    =========================
    Starting. Press "Enter" to stop training and save model.
    Error: operands could not be broadcast together with shapes (144,144,3) (72,72,3) 
    Traceback (most recent call last):
      File "/content/DeepFaceLab/mainscripts/Trainer.py", line 93, in trainerThread
        iter, iter_time = model.train_one_iter()
      File "/content/DeepFaceLab/models/ModelBase.py", line 376, in train_one_iter
        plist += [ (self.get_previews()[0][1], self.get_strpath_storage_for_file('preview.jpg') ) ]
      File "/content/DeepFaceLab/models/ModelBase.py", line 274, in get_previews
        return self.onGetPreview ( self.last_sample )
      File "/content/DeepFaceLab/models/Model_SAE/Model.py", line 411, in onGetPreview
        ar = S[i]*test_S_m[i], SS[i], D[i]*test_D_m[i], DD[i]*DDM[i], SD[i]*(DDM[i]*SDM[i])
    ValueError: operands could not be broadcast together with shapes (144,144,3) (72,72,3) 
    Done.
    /usr/lib/python3.6/multiprocessing/semaphore_tracker.py:143: UserWarning: semaphore_tracker: There appear to be 1 leaked semaphores to clean up at shutdown
      len(cache))
    /usr/lib/python3.6/multiprocessing/semaphore_tracker.py:143: UserWarning: semaphore_tracker: There appear to be 1 leaked semaphores to clean up at shutdown
      len(cache))
    /usr/lib/python3.6/multiprocessing/semaphore_tracker.py:143: UserWarning: semaphore_tracker: There appear to be 1 leaked semaphores to clean up at shutdown
      len(cache))
    /usr/lib/python3.6/multiprocessing/semaphore_tracker.py:143: UserWarning: semaphore_tracker: There appear to be 1 leaked semaphores to clean up at shutdown
      len(cache))
    
    opened by Greywolf22 4
  • Wording/Phrasing changes, Remove References to multiple Drive accounts, Rename function name in Import Workspace, Refactor Dataset/Model Download

    Wording/Phrasing changes, Remove References to multiple Drive accounts, Rename function name in Import Workspace, Refactor Dataset/Model Download

    I saw after the change in Google Drive (same account restriction), that a reference to multiple Google accounts was in the text box, so I ended up removing it but then wanted to go further and try helping with some of the other stuff as well. I updated the GPUs that Colab has as well, filled out the VRAM amount for the GPUs I have gotten. I also changed the function name in "import workspace". Also refactored the dataset and model downloading to hopefully make it easier in the future for multiple options and also allow it to be optional and not required.

    opened by camjac251 10
Releases(pretrain_GenericFFHQ)
Defending graph neural networks against adversarial attacks (NeurIPS 2020)

GNNGuard: Defending Graph Neural Networks against Adversarial Attacks Authors: Xiang Zhang ( Zitnik Lab @ Harvard 44 Dec 07, 2022

A Tensorflow implementation of BicycleGAN.

BicycleGAN implementation in Tensorflow As part of the implementation series of Joseph Lim's group at USC, our motivation is to accelerate (or sometim

Cognitive Learning for Vision and Robotics (CLVR) lab @ USC 97 Dec 02, 2022
Structural Constraints on Information Content in Human Brain States

Structural Constraints on Information Content in Human Brain States Code accompanying the paper "The information content of brain states is explained

Leon Weninger 3 Sep 07, 2022
The 2nd place solution of 2021 google landmark retrieval on kaggle.

Google_Landmark_Retrieval_2021_2nd_Place_Solution The 2nd place solution of 2021 google landmark retrieval on kaggle. Environment We use cuda 11.1/pyt

229 Dec 13, 2022
A annotation of yolov5-5.0

代码版本:0714 commit #4000 $ git clone https://github.com/ultralytics/yolov5 $ cd yolov5 $ git checkout 720aaa65c8873c0d87df09e3c1c14f3581d4ea61 这个代码只是注释版

Laughing 229 Dec 17, 2022
Multi-agent reinforcement learning algorithm and environment

Multi-agent reinforcement learning algorithm and environment [en/cn] Pytorch implements multi-agent reinforcement learning algorithms including IQL, Q

万鲲鹏 7 Sep 20, 2022
Exploration of some patients clinical variables.

Answer_ALS_clinical_data Exploration of some patients clinical variables. All the clinical / metadata data is available here: https://data.answerals.o

1 Jan 20, 2022
This is the official PyTorch implementation of our paper: "Artistic Style Transfer with Internal-external Learning and Contrastive Learning".

Artistic Style Transfer with Internal-external Learning and Contrastive Learning This is the official PyTorch implementation of our paper: "Artistic S

51 Dec 20, 2022
This repository contains the implementation of the paper: "Towards Frequency-Based Explanation for Robust CNN"

RobustFreqCNN About This repository contains the implementation of the paper "Towards Frequency-Based Explanation for Robust CNN" arxiv. It primarly d

Sarosij Bose 2 Jan 23, 2022
PyTorch Implementations for DeeplabV3 and PSPNet

Pytorch-segmentation-toolbox DOC Pytorch code for semantic segmentation. This is a minimal code to run PSPnet and Deeplabv3 on Cityscape dataset. Shor

Zilong Huang 746 Dec 15, 2022
Demonstrates iterative FGSM on Apple's NeuralHash model.

apple-neuralhash-attack Demonstrates iterative FGSM on Apple's NeuralHash model. TL;DR: It is possible to apply noise to CSAM images and make them loo

Lim Swee Kiat 11 Jun 23, 2022
This repository is for EMNLP 2021 paper: It is Not as Good as You Think! Evaluating Simultaneous Machine Translation on Interpretation Data

InterpretationData This repository is for our EMNLP 2021 paper: It is Not as Good as You Think! Evaluating Simultaneous Machine Translation on Interpr

4 Apr 21, 2022
City Surfaces: City-scale Semantic Segmentation of Sidewalk Surfaces

City Surfaces: City-scale Semantic Segmentation of Sidewalk Surfaces Paper Temporary GitHub page for City Surfaces paper. More soon! While designing s

14 Nov 10, 2022
[Preprint] ConvMLP: Hierarchical Convolutional MLPs for Vision, 2021

Convolutional MLP ConvMLP: Hierarchical Convolutional MLPs for Vision Preprint link: ConvMLP: Hierarchical Convolutional MLPs for Vision By Jiachen Li

SHI Lab 143 Jan 03, 2023
Download and preprocess popular sequential recommendation datasets

Sequential Recommendation Datasets This repository collects some commonly used sequential recommendation datasets in recent research papers and provid

125 Dec 06, 2022
2021-AIAC-QQ-Browser-Hyperparameter-Optimization-Rank6

2021-AIAC-QQ-Browser-Hyperparameter-Optimization-Rank6

Aigege 8 Mar 31, 2022
Code to accompany our paper "Continual Learning Through Synaptic Intelligence" ICML 2017

Continual Learning Through Synaptic Intelligence This repository contains code to reproduce the key findings of our path integral approach to prevent

Ganguli Lab 82 Nov 03, 2022
Multi-Modal Machine Learning toolkit based on PyTorch.

简体中文 | English TorchMM 简介 多模态学习工具包 TorchMM 旨在于提供模态联合学习和跨模态学习算法模型库,为处理图片文本等多模态数据提供高效的解决方案,助力多模态学习应用落地。 近期更新 2022.1.5 发布 TorchMM 初始版本 v1.0 特性 丰富的任务场景:工具

njustkmg 1 Jan 05, 2022
A library for uncertainty quantification based on PyTorch

Torchuq [logo here] TorchUQ is an extensive library for uncertainty quantification (UQ) based on pytorch. TorchUQ currently supports 10 representation

TorchUQ 96 Dec 12, 2022
Source code for Zalo AI 2021 submission

zalo_ltr_2021 Source code for Zalo AI 2021 submission Solution: Pipeline We use the pipepline in the picture below: Our pipeline is combination of BM2

128 Dec 27, 2022