PhotoSketch

사진 스케치 : 이미지에서 윤곽 도면을 유추합니다

이 repo에는 스케치 생성기의 교육 및 테스트 코드가 포함되어 있습니다. 우리는 또한 [미리 훈련 된 모델]을 제공합니다.

기술적 인 세부 사항 및 데이터 세트는 [종이] 및 [프로젝트 페이지] 를 참조하십시오.

설정

코드는 이제 Pytorch 0.4를 사용하도록 업데이트되었으며 Windows, Mac 및 Linux에서 실행됩니다. Pytorch 0.3 (Linux 만 해당)이있는 쓸모없는 버전의 경우 Pytorch-0.3-Obsolete 분기를 확인하십시오.

Windows 사용자는 아래에 언급 된 .sh 파일 대신 해당 .cmd 파일을 찾아야합니다.

단선 설치 (콘다 환경 포함)

conda env create -f environment.yml

그런 다음 환경 (스케치)을 활성화하면 갈 준비가되었습니다!

콘다 환경에 대한 자세한 내용은 여기를 참조하십시오.

수동 설치

종속성 목록은 environment.yml 참조하십시오.

미리 훈련 된 모델을 사용합니다

• 미리 훈련 된 모델을 다운로드하십시오
• scripts/test_pretrained.sh 에서 경로를 수정하십시오
• Repo의 루트 디렉토리 에서 scripts/test_pretrained.sh 실행하십시오

입력으로 이미지 폴더를 지원합니다.

컨투어 드로잉 데이터 세트에서 훈련 및 테스트

• 프로젝트 페이지에서 이미지와 렌더링 된 스케치 다운로드
• 압축을 풀고 다음 구조로 구성하십시오.

• scripts/train.sh 및 scripts/test.sh 에서 경로를 수정하십시오
• Repo의 루트 디렉토리 에서 scripts/train.sh 실행하여 모델을 훈련시킵니다.
• Repo의 루트 디렉토리 에서 VAL 세트 또는 테스트 세트에서 테스트를 위해 scripts/test.sh 실행하십시오 (위상 플래그에서 지정 됨)

소환

연구에 코드 또는 데이터를 사용하는 경우 논문을 인용하십시오.

 @article{LIPS2019,
  title={Photo-Sketching: Inferring Contour Drawings from Images},
  author={Li, Mengtian and Lin, Zhe and Mv ech, Radom'ir and and Yumer, Ersin and Ramanan, Deva},
  journal={WACV},
  year={2019}
}

승인

이 코드는 이전 버전의 pix2pix를 기반으로합니다.