Jetson Nano로 StyleGAN 이동

Jetson Nano상에서 StyleGAN이 움직이지 않는다고 하는 코멘트를 보았으므로 「그런 와야겠지」라고 생각해 조속히 시험해 보았습니다. 시도한 코드는 다음과 같습니다.

stylegan(on GitHub)

Jetson Nano 설정

　 이하 기사를 참고로 셋업을 해 주세요.

Jetson Nano 관계의 TIPS 정리

「Swap 파일의 설정」의 항목으로 6GB의 Swap 파일을 확보하는 것을 잊지 말아 주세요(이것을 하지 않으면 도중에 죽습니다).

StyleGAN 테스트

터미널에서 다음 명령을 실행하면 학습된 모델에서 인물 이미지가 생성됩니다. 복제할 리포지토리는 처음 소개한 NVIDIA 오리지널 것일 수 있지만, 이후의 데모 프로그램 실행 관계에서 내가 포크한 것을 사용하고 있습니다.

$ git clone https://github.com/karaage0703/stylegan
$ cd stylegan
$ python3 pretrained_example.py

　 이하와 같은 쾌활한 아저씨 이미지가 생성됩니다. 이것은 교사 이미지가 아니라 인공 지능이 생성한 인물 이미지입니다.

StyleGAN 실시간 생성 데모

그대로 다음 명령 실행하면 차례로 인물 이미지가 생성되는 데모를 할 수 있습니다.

$ python3 generate_demo.py

데모의 모습은, 이하 링크처 참조해 주세요.

Style GAN의 실시간 데모 동영상(on instagram)

데모 프로그램은 기본적으로 샘플 프로그램을 조금 개조한 것입니다. 자세한 내용은 소스 코드를 참조하십시오.

프로그램이 잘 동작하지 않는 경우

모델이 잘 다운로드되지 않은 경우가 있는 것 같습니다. 이 경우 브라우저에서 다음 주소에서 직접 파일을 다운로드하고 stylegan 디렉토리 아래에 직접 놓은 다음 프로그램을 실행하십시오.

참고 링크

A Style-Based Generator Architecture for Generative Adversarial Networks를 읽었습니다.

관련 기사

Jetson Nano 관계의 TIPS 정리
Jetson Nano 관련 정보는 여기에 요약되어 있습니다.

변경 내역

2021/03/01 프로그램이 잘 작동하지 않는 경우의 조치를 추가한다.