Claude said the current codebase (viscy predict) does only rotations, you here I implemented the same. Let's bundle flips in a separate PR? We could package as with_rotation_tta, with_reflection_tta, and with_tta - which allows you to pick between rotation, reflection, or both.
Originally posted by @ieivanov in #450 (comment)
Claude said the current codebase (
viscy predict) does only rotations, you here I implemented the same. Let's bundle flips in a separate PR? We could package aswith_rotation_tta,with_reflection_tta, andwith_tta- which allows you to pick between rotation, reflection, or both.Originally posted by @ieivanov in #450 (comment)