基于深度学习的旧照片着色和修复项目(DeOldify)源码

【实例简介】
DeOldify是一个基于深度学习的项目，用于着色和修复旧照片和视频。你可以免费使用开源的DeOldify来着色图像，有多种工具和平台可供选择，包括Stable Diffusion Web UI插件、ColorfulSoft Windows GUI等。
【实例截图】
【核心代码】
文件清单
└── DeOldify-be725ca6c5f411c47550df951546537d7202c9bc
    ├── ColorFIDBenchmarkArtistic.ipynb
    ├── ColorizeTrainingArtistic.ipynb
    ├── ColorizeTrainingStable.ipynb
    ├── ColorizeTrainingStableLargeBatch.ipynb
    ├── ColorizeTrainingVideo.ipynb
    ├── ColorizeTrainingWandb.ipynb
    ├── deoldify
    │   ├── augs.py
    │   ├── critics.py
    │   ├── dataset.py
    │   ├── device_id.py
    │   ├── _device.py
    │   ├── filters.py
    │   ├── generators.py
    │   ├── __init__.py
    │   ├── layers.py
    │   ├── loss.py
    │   ├── save.py
    │   ├── unet.py
    │   └── visualize.py
    ├── environment.yml
    ├── fastai
    │   ├── basic_data.py
    │   ├── basics.py
    │   ├── basic_train.py
    │   ├── callback.py
    │   ├── callbacks
    │   │   ├── csv_logger.py
    │   │   ├── fp16.py
    │   │   ├── general_sched.py
    │   │   ├── hooks.py
    │   │   ├── __init__.py
    │   │   ├── loss_metrics.py
    │   │   ├── lr_finder.py
    │   │   ├── mem.py
    │   │   ├── misc.py
    │   │   ├── mixup.py
    │   │   ├── mlflow.py
    │   │   ├── one_cycle.py
    │   │   ├── oversampling.py
    │   │   ├── rnn.py
    │   │   ├── tensorboard.py
    │   │   └── tracker.py
    │   ├── collab.py
    │   ├── core.py
    │   ├── data_block.py
    │   ├── datasets.py
    │   ├── distributed.py
    │   ├── gen_doc
    │   │   ├── autogen.tpl
    │   │   ├── convert2html.py
    │   │   ├── core.py
    │   │   ├── docstrings.py
    │   │   ├── doctest.py
    │   │   ├── gen_notebooks.py
    │   │   ├── hide.tpl
    │   │   ├── __init__.py
    │   │   ├── jekyll.tpl
    │   │   ├── nbdoc.py
    │   │   └── nbtest.py
    │   ├── general_optimizer.py
    │   ├── imports
    │   │   ├── core.py
    │   │   ├── __init__.py
    │   │   └── torch.py
    │   ├── __init__.py
    │   ├── launch.py
    │   ├── layers.py
    │   ├── LICENSE
    │   ├── metrics.py
    │   ├── script.py
    │   ├── sixel.py
    │   ├── tabular
    │   │   ├── data.py
    │   │   ├── __init__.py
    │   │   ├── models.py
    │   │   └── transform.py
    │   ├── test_registry.json
    │   ├── text
    │   │   ├── data.py
    │   │   ├── __init__.py
    │   │   ├── interpret.py
    │   │   ├── learner.py
    │   │   ├── models
    │   │   │   ├── awd_lstm.py
    │   │   │   ├── bwd_forget_mult_cuda.cpp
    │   │   │   ├── bwd_forget_mult_cuda_kernel.cu
    │   │   │   ├── forget_mult_cuda.cpp
    │   │   │   ├── forget_mult_cuda_kernel.cu
    │   │   │   ├── __init__.py
    │   │   │   ├── qrnn.py
    │   │   │   └── transformer.py
    │   │   └── transform.py
    │   ├── torch_core.py
    │   ├── train.py
    │   ├── utils
    │   │   ├── check_perf.py
    │   │   ├── collect_env.py
    │   │   ├── __init__.py
    │   │   ├── ipython.py
    │   │   ├── mem.py
    │   │   ├── mod_display.py
    │   │   ├── pynvml_gate.py
    │   │   └── show_install.py
    │   ├── version.py
    │   ├── vision
    │   │   ├── cyclegan.py
    │   │   ├── data.py
    │   │   ├── gan.py
    │   │   ├── image.py
    │   │   ├── __init__.py
    │   │   ├── interpret.py
    │   │   ├── learner.py
    │   │   ├── models
    │   │   │   ├── cadene_models.py
    │   │   │   ├── darknet.py
    │   │   │   ├── __init__.py
    │   │   │   ├── presnet.py
    │   │   │   ├── unet.py
    │   │   │   ├── wrn.py
    │   │   │   ├── xception.py
    │   │   │   ├── xresnet2.py
    │   │   │   └── xresnet.py
    │   │   ├── transform.py
    │   │   └── tta.py
    │   └── widgets
    │       ├── class_confusion.py
    │       ├── image_cleaner.py
    │       ├── image_downloader.py
    │       └── __init__.py
    ├── fid
    │   ├── fid_score.py
    │   ├── inception.py
    │   └── LICENSE
    ├── ImageColorizerArtisticTests.ipynb
    ├── ImageColorizerColab.ipynb
    ├── ImageColorizerColabStable.ipynb
    ├── ImageColorizer.ipynb
    ├── ImageColorizerStableTests.ipynb
    ├── LICENSE
    ├── MANIFEST.in
    ├── models
    ├── README.md
    ├── requirements-colab.txt
    ├── requirements-dev.txt
    ├── requirements.txt
    ├── resource_images
    │   ├── twitter.svg
    │   └── watermark.png
    ├── setup.py
    ├── test_images
    ├── tox.ini
    ├── VideoColorizerColab.ipynb
    └── VideoColorizer.ipynb

17 directories, 138 files