HF RL Explorer

Feature Request: Add Image Moments Computation

Feature Request: Add Image Moments Computation: a task in LegoFlow-SWE (Harbor dataset). When a specific moment is selected (e.g., AF MOMENT M00 ), the function must return an array containing only that moment’s value for each batch/channel. When AF MOMENT FIRST ORDER is selected, the output array…

The task

When a specific moment is selected (e.g., `AF_MOMENT_M00`), the function must return an array containing only that moment’s value for each batch/channel. When `AF_MOMENT_FIRST_ORDER` is selected, the output array must contain all four moments **per batch**, arranged in the order `[M00, M01, M10, M11]`. For a…

Part of Lego-X/LegoFlow-SWE.