-
Notifications
You must be signed in to change notification settings - Fork 1.5k
Amend extract_levels for LocalNet #5542
Copy link
Copy link
Closed
Description
Is your feature request related to a problem? Please describe.
In the current LocalNet implementation, the extract_levels arguments are not directly passed to its parent class RegUNet. Instead, the deepest level max(extract_levels) is passed as the depth argument. Consequentially, the output_block only takes decoded features at the deepest extract level as input.
Describe the solution you'd like
Amend the code to pass the extract_levels arguments to its parent class RegUNet.
Describe alternatives you've considered
N/A
Additional context
N/A
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels