Hello, I want to know that the channel of taxibj dataset is 2. How to calculate the ssim between the predict frames and input frames?