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)
A large-scale video dataset for the training and evaluation of 3D human pose estimation models

ASPset-510 (Australian Sports Pose Dataset) is a large-scale video dataset for the training and evaluation of 3D human pose estimation models. It contains 17 different amateur subjects performing 30

Aiden Nibali 25 Jun 20, 2021
a dnn ai project to classify which food people are eating on audio recordings

Deep Learning - EAT Challenge About This project is part of an AI challenge of the DeepLearning course 2021 at the University of Augsburg. The objecti

Marco Tröster 1 Oct 24, 2021
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
Implementation of the Chamfer Distance as a module for pyTorch

Chamfer Distance for pyTorch This is an implementation of the Chamfer Distance as a module for pyTorch. It is written as a custom C++/CUDA extension.

Christian Diller 205 Jan 05, 2023
[NeurIPS 2021] Source code for the paper "Qu-ANTI-zation: Exploiting Neural Network Quantization for Achieving Adversarial Outcomes"

Qu-ANTI-zation This repository contains the code for reproducing the results of our paper: Qu-ANTI-zation: Exploiting Quantization Artifacts for Achie

Secure AI Systems Lab 8 Mar 26, 2022
Putting NeRF on a Diet: Semantically Consistent Few-Shot View Synthesis Implementation

Putting NeRF on a Diet: Semantically Consistent Few-Shot View Synthesis Implementation This project attempted to implement the paper Putting NeRF on a

254 Dec 27, 2022
Learning to Prompt for Continual Learning

Learning to Prompt for Continual Learning (L2P) Official Jax Implementation L2P is a novel continual learning technique which learns to dynamically pr

Google Research 207 Jan 06, 2023
Repository for the paper "Exploring the Sensory Spaces of English Perceptual Verbs in Natural Language Data"

Sensory Spaces of English Perceptual Verbs This repository contains the code and collocational data described in the paper "Exploring the Sensory Spac

David Peng 0 Sep 07, 2021
PyTorch implementation of Self-supervised Contrastive Regularization for DG (SelfReg)

SelfReg PyTorch official implementation of Self-supervised Contrastive Regularization for Domain Generalization (SelfReg, https://arxiv.org/abs/2104.0

64 Dec 16, 2022
This repository is the code of the paper "Sparse Spatial Transformers for Few-Shot Learning".

🌟 Sparse Spatial Transformers for Few-Shot Learning This code implements the Sparse Spatial Transformers for Few-Shot Learning(SSFormers). Our code i

chx_nju 38 Dec 13, 2022
Towards Long-Form Video Understanding

Towards Long-Form Video Understanding Chao-Yuan Wu, Philipp Krähenbühl, CVPR 2021 [Paper] [Project Page] [Dataset] Citation @inproceedings{lvu2021,

Chao-Yuan Wu 69 Dec 26, 2022
[Pedestron] Generalizable Pedestrian Detection: The Elephant In The Room. @ CVPR2021

Pedestron Pedestron is a MMdetection based repository, that focuses on the advancement of research on pedestrian detection. We provide a list of detec

Irtiza Hasan 594 Jan 05, 2023
Generative Query Network (GQN) in PyTorch as described in "Neural Scene Representation and Rendering"

Update 2019/06/24: A model trained on 10% of the Shepard-Metzler dataset has been added, the following notebook explains the main features of this mod

Jesper Wohlert 313 Dec 27, 2022
The source code for 'Noisy-Labeled NER with Confidence Estimation' accepted by NAACL 2021

Kun Liu*, Yao Fu*, Chuanqi Tan, Mosha Chen, Ningyu Zhang, Songfang Huang, Sheng Gao. Noisy-Labeled NER with Confidence Estimation. NAACL 2021. [arxiv]

30 Nov 12, 2022
Joint detection and tracking model named DEFT, or ``Detection Embeddings for Tracking.

DEFT: Detection Embeddings for Tracking DEFT: Detection Embeddings for Tracking, Mohamed Chaabane, Peter Zhang, J. Ross Beveridge, Stephen O'Hara

Mohamed Chaabane 253 Dec 18, 2022
MCMC samplers for Bayesian estimation in Python, including Metropolis-Hastings, NUTS, and Slice

Sampyl May 29, 2018: version 0.3 Sampyl is a package for sampling from probability distributions using MCMC methods. Similar to PyMC3 using theano to

Mat Leonard 304 Dec 25, 2022
RepMLP: Re-parameterizing Convolutions into Fully-connected Layers for Image Recognition

RepMLP: Re-parameterizing Convolutions into Fully-connected Layers for Image Recognition (PyTorch) Paper: https://arxiv.org/abs/2105.01883 Citation: @

260 Jan 03, 2023
Background Matting: The World is Your Green Screen

Background Matting: The World is Your Green Screen By Soumyadip Sengupta, Vivek Jayaram, Brian Curless, Steve Seitz, and Ira Kemelmacher-Shlizerman Th

Soumyadip Sengupta 4.6k Jan 04, 2023
Unofficial implementation of Perceiver IO: A General Architecture for Structured Inputs & Outputs

Perceiver IO Unofficial implementation of Perceiver IO: A General Architecture for Structured Inputs & Outputs Usage import torch from src.perceiver.

Timur Ganiev 111 Nov 15, 2022
The dataset of tweets pulling from Twitters with keyword: Hydroxychloroquine, location: US, Time: 2020

HCQ_Tweet_Dataset: FREE to Download. Keywords: HCQ, hydroxychloroquine, tweet, twitter, COVID-19 This dataset is associated with the paper "Understand

2 Mar 16, 2022