Great job! I would like to ask how the MASK is generated so that the picture becomes [4, h, w].
Great job! I would like to ask how the MASK is generated so that the picture becomes [4, h, w].