Incorrect indexing for img_std

[akshayjin]

CrossScore/task/core.py

Line 78 in 8d238dd

img_std = self.img_mean_std[None, :3, None, None]

img_mean and img_std are using the same value during input normalization. img_std should be self.img_mean_std[None, 3:, None, None].

[ziruiw-dev]

Hi @akshayjin

Ah thanks for pointing it out! Will fix it!

Best,
Zirui