predict.py 240 字节
Newer Older
H
haoyuying 已提交
1 2 3 4 5 6 7 8 9 10
import paddle
import paddlehub as hub
import paddle.nn as nn


if __name__ == '__main__':
    paddle.disable_static()
    model = hub.Module(directory='user_guided_colorization')
    model.eval()
    result = model.predict(images='sea.jpg')