준비 해야될 리소스 : https://github.com/ZHKKKe/Harmonizer
사용 시스템 : Linux (Ubuntu), Python, Anaconda3
※ CUDA를 사용하는 그래픽 카드
1. 요건
이 데모의 기본 요구 사항은 다음과 같습니다.
- 우분투 시스템
- 파이썬 3+
- 아나콘다3
- openCV
- 파이토치
1-1. 선행설치
- 아나콘다3
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
chmod +x Miniconda3-latest-Linux-x86_64.sh
./Miniconda3-latest-Linux-x86_64.sh
source ~/.bashrc
conda --version
export PATH="/home/username/anaconda3/bin:$PATH"
username에는 사용자 이름을 넣으면 된다.
- openCV
conda install -c conda-forge opencv
- pytorch
conda install pytorch torchvision -c pytorch
2. 데모 실행
다음과 같이 이 데모를 실행하기 위해 새 conda 가상 환경을 만드는 것이 좋습니다.
- 이 저장소를 복제합니다.
git clone https://github.com/ZHKKKe/Harmonizer.git
cd Harmonizer
- 이 링크enhancer.pth 에서 선행 학습된 모델을 다운로드하여 폴더에 넣습니다 .Harmonizer/pretrained/
- 이름이 지정된 conda 가상 환경을 만들고 harmonizer(존재하지 않는 경우) 활성화합니다. python=3.8여기서는 다음을 예로 사용합니다 .
conda create -n harmonizer python=3.8
source activate harmonizer
- PyTorch 및 필요한 Python 종속성을 설치합니다(CUDA 버전이 설치된 PyTorch 버전에서 지원되는지 확인하십시오). 이 저장소의 루트 경로에서 다음을 실행합니다.
pip install -r src/requirements.txt
- 다음과 같이 이 저장소의 루트 경로에서 데모 코드를 실행합니다.여기서 ./demo/video_enhancement/example우리가 제공한 샘플을 포함합니다.
자신의 샘플을 테스트하고 싶다면 폴더를 참조하여 비디오를 ./demo/video_enhancement/example준비하십시오 original.
python -m demo.video_enhancement.run \
--example-path ./demo/video_enhancement/example
- 폴더에서 비디오 향상 결과를 확인하십시오: ./demo/video_enhancement/example/enhanced.
LIST