집 > 기사 > 백엔드 개발 > Python을 사용하여 사진을 애니메이션 스타일 아바타로 변환하세요.

Python을 사용하여 사진을 애니메이션 스타일 아바타로 변환하세요.

王林앞으로: 2023-04-22 22:22:111962검색

최근에 문제가 발생했습니다. 아바타를 애니메이션 스타일로 바꿔야겠다는 생각이 먼저 들었습니다.

Python을 사용하여 사진을 애니메이션 스타일 아바타로 변환하세요.

▲사진을 픽셀 스타일로 변환

아바타를 애니메이션 스타일로 변환하는 것은 원본 이미지 정보와 질감 세부 정보를 유지하면서 실제 사진을 애니메이션/만화 스타일의 비현실적인 이미지로 변환하는 것입니다. 현재 Baidu API 외에도 Github에는 직접 사용할 수 있는 오픈 소스 라이브러리가 많이 있습니다.

그중 AnimeGAN은 우한 대학교와 후베이 기술 대학교의 연구입니다. 이는 신경 스타일 전달 + 생성적 적대 네트워크(GAN)의 조합을 사용하며 그 효과는 우리의 요구와 매우 일치합니다.

Python을 사용하여 사진을 애니메이션 스타일 아바타로 변환하세요.

AnimeGAN은 처음에 Tensorflow 프레임워크를 사용했지만 정보를 쿼리한 결과 해당 프로젝트가 이미 PyTorch 프레임워크를 지원하는 것으로 나타났습니다.

Python을 사용하여 사진을 애니메이션 스타일 아바타로 변환하세요.

주소 : https://github.com/bryandlee/animegan2-pytorch

그리고 예전에 PyTorch를 기반으로 웨이보 댓글 감정분석을 구현한 적이 있어서 사용하는데 부담도 없고 부담도 없습니다. 라이브러리를 설치해야 합니다.

Pytorch 설치

PyTorch는 자연어 처리와 같은 애플리케이션을 위한 Torch 기반의 오픈 소스 Python 기계 학습 라이브러리입니다. 이 딥러닝 프레임워크는 수치 모델링, 이미지 모델링, 텍스트 모델링, 오디오 모델링 등 다양한 방향으로 적용될 수 있습니다.

Python을 사용하여 사진을 애니메이션 스타일 아바타로 변환하세요.

Pytorch 설치는 다른 라이브러리에 비해 번거롭습니다. 정식 설치에 가서 다운로드를 하게 되면 실제 구성에 맞는 설치 명령을 받아야 합니다.

Python을 사용하여 사진을 애니메이션 스타일 아바타로 변환하세요.

위 단계를 통해 다운로드 속도가 느리거나 다양한 문제가 발생하는 경우 다음 웹사이트를 시도해 볼 수도 있습니다:

https://download.pytorch.org/whl/torch_stable.html

웹사이트 보기:

Python을 사용하여 사진을 애니메이션 스타일 아바타로 변환하세요.

애니메이션 스타일 마이그레이션

Pytorch 프레임워크를 설치한 후 animationgan2-pytorch 프로젝트를 로컬로 복제하거나 직접 다운로드할 수 있습니다.

git clone https://github.com/bryandlee/animegan2-pytorch

아래와 같이 로컬 최종 디렉터리에 다운로드합니다.

Python을 사용하여 사진을 애니메이션 스타일 아바타로 변환하세요.

가중치 폴더에는 원하는 애니메이션 스타일 마이그레이션을 달성하기 위해 해당 가중치를 선택합니다. 그림 예제는 샘플 아래의 입력 폴더에 저장되어 있으며, 이는 물을 테스트하는 데 직접 사용할 수 있습니다. 또한 처리된 이미지를 저장하기 위해 동일한 경로 아래에 새 출력 폴더도 만들었습니다.

다음으로 프로젝트를 호출하려면 명령줄에서 test.py 스크립트만 실행하면 됩니다. 구체적인 명령 형식은 다음과 같습니다.

python test.py --checkpoint [모델 파일 경로] --input_dir [디렉토리] 입력 이미지가 있는 위치] ] --output_dir [출력 디렉터리] --device [장치 선택, CPU 또는 Cuda]

실제 동작 :

Python을 사용하여 사진을 애니메이션 스타일 아바타로 변환하세요.