Skip to content

Conversation

@talonchandler
Copy link
Collaborator

This PR factors out a waveorder.focus.compute_midband_power function.

This function takes and returns torch.Tensor objects to enable this function to be used as a torch loss.

@gav-sturm, I've written this PR to address your needs for OPS. Please let me know if you need additional changes.
@ieivanov, I have not changed the functionality (or tests) of the focus_from_transverse_band function, so I expect mantis' focus finding to continue working smoothly.

Copy link
Collaborator

@gav-sturm gav-sturm left a comment

Choose a reason for hiding this comment

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

Exactly what we need in OPS. Approved!

@gav-sturm gav-sturm merged commit 96872f5 into main Sep 2, 2025
11 checks passed
@ieivanov
Copy link
Contributor

ieivanov commented Sep 3, 2025

Thanks for flagging this Talon! shrimPy currently depends on the return_peak_stats branch - we should clean that up in the future, for now this PR doesn't cause issues for me, and thanks for looking out for that.

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

Successfully merging this pull request may close these issues.

4 participants