My Data Loader :
LoadData(Dataset): def __init__(self, ..., ...., ...): self.ns = (640, 640) ..... ..... def __getitem__(index