PhotoSketch

Photo-esgotamento: Inferindo desenhos de contorno de imagens

Este repositório contém o código de treinamento e teste para o nosso gerador de esboço. Também fornecemos um [modelo pré-treinado].

Para detalhes técnicos e o conjunto de dados, consulte o [Paper] e a [página do projeto] .

Configurando

O código agora está atualizado para usar o Pytorch 0.4 e é executado no Windows, Mac e Linux. Para a versão obsoleta com Pytorch 0.3 (somente Linux), consulte o ramo Pytorch-0.3-OBSOLETE.

Os usuários do Windows devem encontrar os arquivos .cmd correspondentes em vez dos arquivos .sh mencionados abaixo.

Instalação de uma linha (com ambientes condá)

conda env create -f environment.yml

Em seguida, ative o ambiente (esboço) e você está pronto para ir!

Veja aqui para obter mais informações sobre os ambientes do CONDA.

Instalação manual

Consulte environment.yml para obter uma lista de dependências.

Usando o modelo pré-treinado

• Baixe o modelo pré-treinado
• Modifique o caminho em scripts/test_pretrained.sh
• No diretório raiz do repo, execute scripts/test_pretrained.sh

Ele suporta uma pasta de imagens como entrada.

Trem e teste em nosso conjunto de dados de desenho de contorno

• Baixe as imagens e o esboço renderizado da página do projeto
• Descompacione -os e organize -os na seguinte estrutura:

• Modifique o caminho em scripts/train.sh e scripts/test.sh
• No diretório raiz do repo, execute scripts/train.sh para treinar o modelo
• No diretório raiz do repo, execute scripts/test.sh para testar no conjunto Val ou no conjunto de testes (especificado pelo sinalizador de fase)

Citação

Se você usar o código ou os dados para sua pesquisa, cite o artigo:

 @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}
}

Reconhecimento

Este código é baseado em uma versão antiga do PIX2PIX.