转载自:
https://www.jianshu.com/p/bb90bff9f6e5
总之,collate_fn就是,在若干张图片拼接为一个batch的时候,我们需要组织一种数据拼接方式,这个数据拼接方式就是collate_fn。这里说明了batch当中每个data由哪些元素组成
转载自:
https://www.jianshu.com/p/bb90bff9f6e5
总之,collate_fn就是,在若干张图片拼接为一个batch的时候,我们需要组织一种数据拼接方式,这个数据拼接方式就是collate_fn。这里说明了batch当中每个data由哪些元素组成