IDKiro / StarEnhancer

[ICCV 2021 Oral] StarEnhancer: Learning Real-Time and Style-Aware Image Enhancement
MIT License
150 stars 15 forks source link

About pretrain model #12

Closed hhhhnwl closed 1 year ago

hhhhnwl commented 1 year ago

提供的pretrain模型可以直接跑出enhancer的效果吗,还是说必须先加载进来训练呢,我发现直接加载pretrain的模型跑出的图片没有什么效果。

IDKiro commented 1 year ago

没太搞懂你的问题。 提供的模型权重准确来讲就是训练好的模型,请确保文件放在正确的路径并进行加载。 注意模型分为四个部分,测试时至少要加载enhancer、mapping的网络权重以及风格表征的embedding文件。