I would recommend to test some already implemented approaches such as pytorch-fid 5 and check if they would be working for your use case. Also, I think you might be able to find some implementations for these metrics in this forum.