Skip to content

Add U-Net and draft of Deep Image Prior. Some refactoring #2

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.

Sign up for GitHub

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

Jump to bottom
Merged
merged 12 commits into from
Apr 7, 2025
Merged

Add U-Net and draft of Deep Image Prior. Some refactoring #2

merged 12 commits into from
Apr 7, 2025

Conversation

Melvin-klein
Copy link
Member

Tasks:

  • Add debluring task

Solvers :

  • Add U-Net
  • Add Deep Image Prior (not yet finished)

Refactoring:

  • Removed constants file from benchmark_utils

agramfort
agramfort reviewed Mar 11, 2025
View reviewed changes
tomMoral
tomMoral reviewed Apr 4, 2025
View reviewed changes
solvers/u-net.py
optimizer=optimizer,
show_progress_bar=True, # disable progress bar for better vis in sphinx gallery.
train_dataloader=self.train_dataloader,
)
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Add early stopping

@Melvin-klein Melvin-klein merged commit 0cbadcf into main Apr 7, 2025
4 checks passed
@Melvin-klein Melvin-klein deleted the unet-and-dip branch April 8, 2025 07:33
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@agramfort agramfort agramfort left review comments

@tomMoral tomMoral tomMoral left review comments

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
3 participants
@Melvin-klein @agramfort @tomMoral