It would be nice to flatten so we get per class AP numbers. This doesn't work right now though and just returns the average 🙃 _Originally posted by @dxoigmn in https://github.com/IntelLabs/MART/pull/192#discussion_r1259181499_