转载自：
https://www.jianshu.com/p/bb90bff9f6e5
总之，collate_fn就是，在若干张图片拼接为一个batch的时候，我们需要组织一种数据拼接方式，这个数据拼接方式就是collate_fn。这里说明了batch当中每个data由哪些元素组成
pytorch当中的dataloader的collate_fn的使用
猜你喜欢
转载自blog.csdn.net/mdjxy63/article/details/85345988
今日推荐
周排行