why is my generated result image is also gray?

[honeytidy]

I am running with the test code and got all result also grays. It is like that:

My pytorch vision is 1.5.

[ericsujw]

Do you follow the setting in the colab notebook?
You can run sh scripts/test_mask.sh to get colorized results.

[honeytidy]

Yes, I do. infer box first and then run fusion. But I did this on my windows and cpu model. Do it matter.
And I as you can see that color seems on a small area. How to debug that?

[thelastlin]

Yes, I do. infer box first and then run fusion. But I did this on my windows and cpu model. Do it matter.
And I as you can see that color seems on a small area. How to debug that?

Please have a look at PR #12. The pretrained weights may not be loaded.

[delldu]

On Linux, CPU is also too, it is very strange, but GPU works well.

[sysuzyq]

https://github.com/ericsujw/InstColorization/blob/master/models/fusion_model.py#L108

        GF_state_dict_temp ={}
        for k,v in GF_state_dict.items():
            if k.startswith('module'):
                GF_state_dict_temp[k[7:]] = v
            else:
                GF_state_dict_temp[k] = v
        GF_state_dict = GF_state_dict_temp

        G_state_dict_temp ={}
        for k,v in G_state_dict.items():
            if k.startswith('module'):
                G_state_dict_temp[k[7:]] = v
            else:
                G_state_dict_temp[k] = v
        G_state_dict = G_state_dict_temp

        GComp_state_dict_temp ={}
        for k,v in GComp_state_dict.items():
            if k.startswith('module'):
                GComp_state_dict_temp[k[7:]] = v
            else:
                GComp_state_dict_temp[k] = v
        GComp_state_dict = GComp_state_dict_temp

        self.netGF.load_state_dict(GF_state_dict, strict=True)
        self.netG.load_state_dict(G_state_dict, strict=True)
        self.netGComp.load_state_dict(GComp_state_dict, strict=True)