/Reference-based-colorization-rest-api

Rest API를 활용한 딥러닝 모델(Reference-Based-Colorization) 서빙

Primary LanguagePython

Backend

Pretrained model

put colornet_iter_76000.pth in /checkpoints/video_moredata_l1

put vgg19_conv.pth, vgg19_gray.pth in /colorization/data

Requirements

pip install -r requirements-{OS}.txt

Run server

gunicorn --bind 0.0.0.0:5000 wsgi:app

How to use

Post at {domain}/colorization

with body including

ref_id      => integer(1~10)

image       => Image

ref_option  => integer(0~1, 0:original ref, 1:user ref)

user_ref    => Image

Response : Reference-based colorized image

Result

image