Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

is this model can be used to do image deburing ? #1

Open
sunzhe09 opened this issue Feb 18, 2024 · 4 comments
Open

is this model can be used to do image deburing ? #1

sunzhe09 opened this issue Feb 18, 2024 · 4 comments

Comments

@sunzhe09
Copy link

No description provided.

@zzr-idam
Copy link
Owner

zzr-idam commented Feb 18, 2024 via email

@sunzhe09
Copy link
Author

HI,how to test with my custom blurry image?

@zzr-idam
Copy link
Owner

We provide the training code, first this model should be trained on your customized dataset and then tested.

@zzr-idam
Copy link
Owner

There is no need to reduce the number of layers in the UNet, my basic module has three SSMs, you can remove the one that is the most computationally intensive (after reshaping); besides, you can conduct a ReLU or Sigmoid on the output of the UNet and do a residual or multiplication operation with the original input image.

output = nn.ReLU()(UNet(x))

output = output * x - output + 1 // AODNet equation

return output

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants