OSCHINA-MIRROR/open-mmlab-mmgeneration

OpenMMLab website

OpenMMLab platform

Документация, Установка, Зоопарк моделей, Новости об обновлениях, Текущие проекты, Сообщение о проблемах.

Английский | 简体中文

Что нового

MMGeneration был объединён с MMEditing. Мы добавили поддержку новых задач и моделей для генерации. Вот основные новые функции:

🌟 Генерация текста в изображение:
- ✅ GLIDE;
- ✅ Disco-Diffusion;
- ✅ Stable-Diffusion.
🌟 3D-осведомлённая генерация:
- ✅ EG3D.

Введение

MMGeneration — это мощный инструментарий для генеративных моделей, особенно для GAN. Он основан на PyTorch и MMCV. Основная ветка работает с PyTorch 1.5+.

Основные функции

Высокое качество обучения: в настоящее время мы поддерживаем обучение для безусловных GAN, внутренних GAN и моделей перевода изображений. Поддержка условных моделей появится в ближайшее время.
Мощный инструментарий приложений: пользователям предоставляется обширный инструментарий, содержащий множество приложений для GAN. В нашу структуру интегрированы интерполяция GAN, проекция GAN и манипуляции GAN. Пришло время поиграть с вашими GAN! (Учебник по приложениям).
Эффективное распределённое обучение для генеративных моделей: для высокодинамичного обучения в генеративных моделях мы применяем новый способ обучения динамических моделей с помощью MMDDP. (Учебник по DDP).
Новый модульный дизайн для гибкой комбинации: предложен новый дизайн для сложных модулей потерь для настройки связей между модулями, что позволяет гибко комбинировать различные модули. (Учебник по новому модульному дизайну). Связанные приложения:

MMGen-FaceStylor (https://github.com/open-mmlab/MMGEN-FaceStylor).

Вклад:

Мы ценим любой вклад в улучшение MMGeneration. Пожалуйста, обратитесь к CONTRIBUTING.md в MMCV для получения более подробной информации о руководстве по внесению вклада.

Цитирование:

Если вы считаете этот проект полезным в своём исследовании, пожалуйста, рассмотрите возможность цитирования:

@misc{2021mmgeneration,
    title={{MMGeneration}: OpenMMLab Generative Model Toolbox and Benchmark},
    author={MMGeneration Contributors},
    howpublished = {\url{https://github.com/open-mmlab/mmgeneration}},
    year={2021}
}

Лицензия:

Этот проект выпущен под лицензией Apache 2.0 (LICENSE). Некоторые операции в MMGeneration выполняются с другими лицензиями вместо Apache2.0. Пожалуйста, обратитесь к LICENSES.md для тщательной проверки, если вы используете наш код в коммерческих целях.

Проекты в OpenMMLab:

— MMCV (https://github.com/open-mmlab/mmcv): основополагающая библиотека OpenMMLab для компьютерного зрения. — MIM (https://github.com/open-mmlab/mim): MIM устанавливает пакеты OpenMMLab. — MMClassification (https://github.com/open-mmlab/mmclassification): набор инструментов и эталонный тест классификации изображений OpenMMLab. — MMDetection (https://github.com/open-mmlab/mmdetection): набор инструментов обнаружения и эталонный тест OpenMMLab. — MMDetection3D (https://github.com/open-mmlab/mmdetection3d): платформа следующего поколения OpenMMLab для общего обнаружения объектов в 3D. — MMRotate (https://github.com/open-mmlab/mmrotate): набор инструментов для обнаружения повёрнутых объектов и эталонный тест OpenMMLab. — MMSegmentation (https://github.com/open-mmlab/mmsegmentation): набор инструментов семантической сегментации и эталонный тест OpenMMLab. — MMOCR (https://github.com/open-mmlab/mmocr): обнаружение текста, распознавание и понимание текста OpenMMLab. — MMPose (https://github.com/open-mmlab/mmpose): набор инструментов оценки позы и эталонный тест OpenMMLab. — MMHuman3D (https://github.com/open-mmlab/mmhuman3d): набор инструментов 3D-модели человека и эталонный тест OpenMMLab. — MMSelfSup (https://github.com/open-mmlab/mmselfsup): набор инструментов самообучения и эталонный тест OpenMMLab. — MMRazor (https://github.com/open-mmlab/mmrazor): набор инструментов сжатия модели и эталонный тест OpenMMLab. — MMFewShot (https://github.com/open-mmlab/mmfewshot): набор инструментов обучения на основе нескольких примеров и эталонный тест OpenMMLab. — MMAction2 (https://github.com/open-mmlab/mmaction2): набор инструментов понимания действий следующего поколения и эталонный тест OpenMMLab. — MMTracking (https://github.com/open-mmlab/mmtracking): набор инструментов восприятия видео и эталонный тест OpenMMLab. — MMFlow (https://github.com/open-mmlab/mmflow): набор инструментов оптического потока и эталонный тест OpenMMLab. — MMEditing (https://github.com/open-mmlab/mmediting): набор инструментов редактирования изображений и видео OpenMMLab. — MMGeneration (https://github.com/open-mmlab/mmgeneration): набор инструментов генеративных моделей изображений и видео OpenMMLab. — MMDeploy (https://github.com/open-mmlab/mmdeploy): среда развёртывания моделей OpenMMLab.