normol的博客

前言：第一次接触视频分割相关的网络(RVOS,https://github.com/imatge-upc/rvos)，发现网络dataloader的输出数据格式不是batch first，而是[length_of_clips, batch_size, channels, height, width]。经过实验发现：若Dataset返回tensor对象，则dataloader输出常规的[batch_size, length_of_clips, channels, height, width]，(即batch