richzhang / colorization-pytorch

PyTorch reimplementation of Interactive Deep Colorization

Home Page:https://richzhang.github.io/ideepcolor/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Convert into ONNX

97mik opened this issue · comments

How can I export ONNX model from latest_net_G.pth?

Not sure how to do it, but a more straightforward model file just for execution is in the GUI repository, more specifically here: https://github.com/junyanz/interactive-deep-colorization/blob/master/models/pytorch/model.py

I have the same question. How can I load your pretrained model from latest_net_G.pth as nn.module model? The script you refered to is only defining the network structure. Thank you so much.

See test.py and this line is loading the model:

model = create_model(opt)