이 버전은 이전 버전과 호환되지 않습니다. 이전 버전을 사용하려면 old_version
브랜치를 참고하세요.
영어 |简体中文 GitHub | Gitee码云
Google Colab: GitHub 링크 | 구글 드라이브 링크
? 데이터 세트 ⏬ Google 드라이브 ⏬ 百島网盘 (提取码:basr)
? wandb의 훈련 곡선
훈련 및 테스트를 위한 명령
⚡ 노하우
BasicSR( Basic Super Restoration )은 초고해상도, 노이즈 제거, 디블러링, JPEG 아티팩트 제거 등과 같은 PyTorch 기반의 오픈 소스 이미지 및 비디오 복원 도구 상자입니다.
(ESRGAN, EDVR, DNI, SFTGAN) (HandyView, HandyFigure, HandyCrawler, HandyWriting)
우리는 빠른 시작을 위해 모델을 훈련/테스트/추론할 수 있는 간단한 파이프라인을 제공합니다. 이러한 파이프라인/명령은 모든 사례를 다룰 수 없으며 자세한 내용은 다음 섹션에 나와 있습니다.
간 | |||||
---|---|---|---|---|---|
스타일GAN2 | 기차 | 추론 | |||
얼굴 복원 | |||||
DFDNet | - | 추론 | |||
초해상도 | |||||
에스르간 | TODO | TODO | 스르간 | TODO | TODO |
EDSR | TODO | TODO | SRResNet | TODO | TODO |
RCAN | TODO | TODO | |||
EDVR | TODO | TODO | DUF | - | TODO |
기본VSR | TODO | TODO | TOF | - | TODO |
디블러링 | |||||
DeblurGANv2 | - | TODO | |||
노이즈 제거 | |||||
RIDNet | - | TODO | CBDNet | - | TODO |
복제 저장소
git clone https://github.com/xinntao/BasicSR.git
종속 패키지 설치
cd BasicSR
pip install -r requirements.txt
BasicSR 설치
BasicSR을 설치하려면 BasicSR 루트 경로 에서 다음 명령을 실행하십시오.
(GCC 버전이 gcc >= 5인지 확인하세요.)
cuda 확장이 필요하지 않은 경우:
EDVR용 dcn
StyleGAN2용 upfirdn2d 및 fused_act
설치 시 --no_cuda_ext
추가하세요.
python setup.py develop --no_cuda_ext
EDVR 및 StyleGAN2 모델을 사용하는 경우 위의 cuda 확장이 필요합니다.
python setup.py develop
CUDA 경로를 지정할 수도 있습니다.
CUDA_HOME=/usr/local/cuda
CUDNN_INCLUDE_DIR=/usr/local/cuda
CUDNN_LIB_DIR=/usr/local/cuda
python setup.py develop
BasicSR은 Ubuntu에서만 테스트되었으며 Windows에는 적합하지 않을 수 있습니다. CUDA를 지원하는 Windows WSL을 사용해 볼 수 있습니다. :-) (이제는 Fast Ring을 사용하는 내부자 빌드에만 사용할 수 있습니다.)
프로젝트 보드를 참조하세요.
torch.utils.data.Dataset
클래스)에 대한 설명은 Datasets.md에 있습니다. BasicSR 코드베이스의 디자인 및 규칙은 DesignConvention.md를 참조하세요.
아래 그림은 전체적인 프레임워크를 보여줍니다. 각 구성요소에 대한 추가 설명:
데이터세트.md | 모델즈.md | 구성.md | 로깅.md
이 프로젝트는 Apache 2.0 라이센스에 따라 릴리스됩니다.
라이센스 및 승인 에 대한 자세한 내용은 LICENSE에 나와 있습니다.
BasicSR이 귀하의 연구나 업무에 도움이 된다면 BasicSR 인용을 고려해 보세요.
다음은 BibTeX 참조입니다. BibTeX 항목에는 url
LaTeX 패키지가 필요합니다.
@misc{wang2020basicsr,
author = {Xintao Wang and Ke Yu and Kelvin C.K. Chan and
Chao Dong and Chen Change Loy},
title = {BasicSR},
howpublished = { url {https://github.com/xinntao/BasicSR}},
year = {2020}
}
Xintao Wang, Ke Yu, Kelvin CK Chan, Chao Dong 및 Chen Change Loy. BasicSR. https://github.com/xinntao/BasicSR, 2020.
질문이 있으시면 [email protected]
으로 이메일을 보내주세요.