mmgeneration 다운로드 mmgeneration 소스 코드 다운로드

mmgeneration

파이썬

v0.7.3 Released

다운로드

영어 | 简体中文

새로운 것

mmgeneration은 mmediting에서 병합되었습니다. 그리고 우리는 새로운 세대의 과제와 모델을 지원했습니다. 우리는 다음과 같은 새로운 기능을 강조합니다.

? Text2image
- ✅ 미끄러짐
- ✅ 디스코 확산
- ✅ 안정된 확산
? 3D 인식 세대
- EG3D

소개

MMGeneration은 생성 모델, 특히 GANS의 강력한 툴킷입니다. Pytorch 및 MMCV를 기반으로합니다. 마스터 브랜치는 Pytorch 1.5+ 와 함께 작동합니다.

주요 기능

고품질 교육 성능 : 우리는 현재 무조건적인 간, 내부 GAN 및 이미지 번역 모델에 대한 교육을 지원합니다. 조건부 모델에 대한 지원은 곧 올 것입니다.
강력한 애플리케이션 툴킷 : GAN에 여러 응용 프로그램이 포함 된 풍부한 툴킷이 사용자에게 제공됩니다. Gan 보간, GAN 프로젝션 및 GAN 조작이 우리의 프레임 워크에 통합됩니다. 당신의 간과 놀야 할 시간입니다! (응용 프로그램 튜토리얼)
생성 모델을위한 효율적인 분산 교육 : 생성 모델에서 매우 역동적 인 교육을 위해 MMDDP 로 동적 모델을 훈련시키는 새로운 방법을 채택합니다. (DDP 튜토리얼)
유연한 조합을위한 새로운 모듈 식 디자인 : 복잡한 손실 모듈을위한 새로운 설계는 모듈 간의 링크를 사용자 정의하기 위해 제안되어 있으며, 이는 다른 모듈간에 유연한 조합을 달성 할 수 있습니다. (새로운 모듈 식 디자인 튜토리얼)

교육 시각화	간 보간	간 프로젝터	간 조작

가장 밝은 부분

GAN의 공간 유도 바이어스 (CVPR2021)로서의 위치 인코딩은 MMGeneration 에서 방출되었다. [구성], [프로젝트 페이지]
조건부가 툴킷에서 지원되었습니다. 더 많은 방법과 미리 훈련 된 무게가 곧 올 것입니다.
StyleGan2에 대한 혼합-정밀 훈련 (FP16)이 지원되었습니다. 다른 구현 간의 비교를 확인하십시오.

changelog

V0.7.3은 20123 년 4 월 4 일에 출시되었습니다. 자세한 내용 및 릴리스 기록은 Changelog.md를 참조하십시오.

설치

mmgeneration은 pytorch 및 mmcv에 따라 다릅니다. 다음은 설치를위한 빠른 단계입니다.

1 단계. 공식 지침에 따라 Pytorch를 설치하십시오

 pip3 install torch torchvision

2 단계. MIM으로 MMCV를 설치하십시오.

 pip3 install openmim
mim install mmcv-full

3 단계. 소스에서 mmgeneration을 설치하십시오.

 git clone https://github.com/open-mmlab/mmgeneration.git
cd mmgeneration
pip3 install -e .

자세한 지침은 get_started.md를 참조하십시오.

시작하기

mmgeneration의 기본 사용에 대해서는 get_started.md를 참조하십시오. Docs/en/Quick_run.md는 Quick Run을위한 전체 지침을 제공 할 수 있습니다. 다른 세부 사항과 튜토리얼은 문서를 참조하십시오.

Modelzoo

이러한 방법은 우리의 프레임 워크에서 신중하게 연구되고 지원되었습니다.

무조건 간 (무너지기 위해 클릭)

DCGAN (ICLR'2016)
WGAN-GP (NIPS'2017)
LSGAN (ICCV'2017)
✅ GGAN (Arxiv'2017)
PGGAN (ICLR'2018)
Styleganv1 (CVPR'2019)
✅ Styleganv2 (CVPR'2020)
Styleganv3 (Neurips'2021)
✅ Gans의 위치 인코딩 (CVPR'2021)

조건부 간 (붕괴 클릭)

SNGAN (ICLR'2018)
Projection Gan (ICLR'2018)
Sagan (ICML'2019)
biggan/biggan-deep (ICLR'2019)

GANS의 트릭 (클릭하여 붕괴)

ADA (Neurips'2020)

image2image 번역 (붕괴 클릭)

pix2pix (CVPR'2017)
Cyclegan (ICCV'2017)

내부 학습 (붕괴 클릭)

Singan (ICCV'2019)

비난 확산 확률 모델 (붕괴 클릭)

ddpm 개선 (Arxiv'2021)

기여

Mmgeneration을 개선하기위한 모든 기여에 감사드립니다. 기여 가이드 라인에 대한 자세한 내용은 MMCV의 Contributing.md를 참조하십시오.

소환

이 프로젝트가 연구에 유용하다고 생각되면 인용을 고려하십시오.

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