ugness / line-art-colorization-spade Goto Github PK

View Code? Open in Web Editor NEW

13.0 1.0 1.0 24.54 MB

SPADE-based Line Art Colorization

License: Other

Python 99.82% Shell 0.18%

colorization spade computer-vision style-transfer safebooru demo deep-learning pytorch

line-art-colorization-spade's People

Contributors

Stargazers

Watchers

Forkers

hehesam

line-art-colorization-spade's Issues

Let's Do Training!!!!

첫 실험 결과

색 번짐.

Milestone + Issue 관리하는 법

Milestone 관리하기

자신이 할 Task를 n개의 세부목표로 나눈다.
n개의 세부목표에 대하여 Issue를 생성한다.
해당 Issue에 해당 Milestone을 지정한다.
해당 목표를 완료하면 해당 이슈를 Close한다.

Assignee 지정하기

Safebooru Dataset

https://www.kaggle.com/alamson/safebooru/download

Xdog Sketch Simplification

https://github.com/heitorrapela/xdog
원본 이미지 사이즈 유지할 것.
(Optional Data)

Demo features

Lineart 이미지 업로드 가능해야함
User hint 마우스로 줄 수 있어야함. 색상 팔레트 이용
Color Image 다운로드 가능해야함.
서버로 접속 가능하면 좋음

더 필요한 demo feature들 혹시 있음??

근데 color image가 다운로드 가능해야된다는건 채색 완료 된 image가 다운로드 할 수 있어야한다는거지?

Android로 짜는것보다 웹으로하는게 낫겠지?

Dataloader Input / Output 설계 (init, getitem)

torch.utils.data.Dataset 모듈 사용할 것.

Dataset 폴더 경로 받아서 init
Normalized Lineart/Hint/Target 을 내뱉음. getitem

SketchKeras로 Sketch 추출하기

원본이미지 사이즈 유지할 것.
https://github.com/lllyasviel/sketchKeras

Inference 코드 작성하기

Demo에 사용할 것 고려해서 작성하기.

Sketch Simplification

https://github.com/bobbens/sketch_simplification
원본 이미지 사이즈 유지할 것

Demo

Lineart 이미지 업로드 가능해야함
User hint 마우스로 줄 수 있어야함. 색상 팔레트 이용
Color Image 다운로드 가능해야함.
서버로 접속 가능하면 좋음

Mean and Stdev of Our Dataset / Normalize

find Mean and Stdev of Our Dataset and use them to normalize

(x - mean)/stdev

Image Download Multithread

image_downloader.py 이미지 다운로드 속도 최대화하기

RGB Image to Lab Image / Lab to RGB Image

왜 LAB 색공간인가

https://ko.wikipedia.org/wiki/CIELAB_%EC%83%89_%EA%B3%B5%EA%B0%84

Feature Request

필요한 Feature 자유롭게 적어주셈

Tensorboard logging
Hint Visualization
연한 컬러 해결하기

How to Download Dataset

Safebooru csv download

https://www.kaggle.com/alamson/safebooru/download 에서 dataset 다운로드.
utils/tag_parser.py 이용해서 필요한 태그만 걸러내기.

usage: tag_parser.py [-h] [--savedir SAVEDIR] [--metadata *.csv]
                     [--tag_list TAG_LIST [TAG_LIST ...]]
                     [--ignore [IGNORE [IGNORE ...]]]

Download images from given metadata.

optional arguments:
  -h, --help            show this help message and exit
  --savedir SAVEDIR     directory to save parsed_metadata
  --metadata *.csv      csv file which contains metadata(urls)
  --tag_list TAG_LIST [TAG_LIST ...]
                        Tags that you want to parse
  --ignore [IGNORE [IGNORE ...]]
                        Tags that you do not want to parse

example : tag_parser.py --savedir safebooru --metadata all_data.csv --tag_list solo white_background tag1 tag2 ... --ignore chibi monster pokemon ....

utils/image_downloader.py 이용해서 다운로드 받기. (절반은 URL 불량)

usage: image_downloader.py [-h] [--savedir SAVEDIR] [--metadata *.csv] [--size SIZE]

Download images from given metadata.

optional arguments:
  -h, --help         show this help message and exit
  --savedir SAVEDIR  directory to save images
  --metadata *.csv   csv file which contains metadata(urls)
  --size SIZE     max length of downloaded images (defualt: Original Image size)

대충 10만장 -> 30GB.
tag parsing 할 때 solo white_background는 필수로 넣기.

Repo 정리할 시 참고

https://github.com/victoresque/pytorch-template

Training Scheme 구성 + 학습 시작

Recommendation
Optimizer RAdam 써보기?
https://github.com/LiyuanLucasLiu/RAdam