can't generate reasonable result

[CDJ-happier]

I git clone this repository, and run the demo_decode.py, i get a picture with much nosie, which totaly can't figure out the image content.
I try above operation on both 2070 and 3090, get same issues

[Shzhuoyu]

I have same question, do you get image like this?

[CDJ-happier]

yes, totally same Cai ***@***.*** Original Email Sender:"Shzhuoyu"< ***@***.*** &gt;; Sent Time:2023/5/22 20:40 To:"AugmentariumLab/SIGNET"< ***@***.*** &gt;; Cc recipient:"CofCai"< ***@***.*** &gt;;"Author"< ***@***.*** &gt;; Subject:Re: [AugmentariumLab/SIGNET] can&#39;t generate reasonable result (Issue#3) I have same question, do you get image like this? &mdash; Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: ***@***.***&gt;

[Shzhuoyu]

I solved it! add these three lines in func "get_LF_val"

val_inp_t[..., :2] /= 17
val_inp_t[..., 2] /= width
val_inp_t[..., 3] /= height

[CDJ-happier]

nice, I will try it. Cai ***@***.*** Original Email Sender:"Shzhuoyu"< ***@***.*** &gt;; Sent Time:2023/5/23 20:13 To:"AugmentariumLab/SIGNET"< ***@***.*** &gt;; Cc recipient:"Cai Dongjun"< ***@***.*** &gt;;"Author"< ***@***.*** &gt;; Subject:Re: [AugmentariumLab/SIGNET] can&#39;t generate reasonable result (Issue#3) I solved it! add these three lines in func "get_LF_val" &mdash; Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: ***@***.***&gt;

[CDJ-happier]

I solved it! add these three lines in func "get_LF_val"

val_inp_t[..., :2] /= 17
val_inp_t[..., 2] /= width
val_inp_t[..., 3] /= height

It doesn't work by adding the three line codes, I don't know why. I got a picture with all pixels are same color by your approaches which is different from the previous result(nosiy images)

[dli7319]

I think the layernorm parameters somehow got turned off in the demo code.
Can you try the fix_demo_decode branch to see if it is fixed?