Perhaps you need to apply the transform on the mask too (excluding normalization). Like class CustomDataset(Dataset): def __init__(self, ...
確定! 回上一頁