search

xiaolai-sqlai / gluon-reid

A code gallery for person re-identification with mxnet-gluon, and I will reproduce many STOA algorithm.
176 stars 28 forks source link

RandomIdentitySampler #1

Open kli-casia opened 6 years ago

kli-casia commented 6 years ago

https://github.com/xiaolai-sqlai/gluon-reid/blob/master/branch/data_read/sampler.py#L35 这里的len是不是应该返回self.num_samples？也就是_iter_函数中ret的长度

我看https://mxnet.incubator.apache.org/_modules/mxnet/gluon/data/sampler.html#BatchSampler batchsampler的源码返回的是batch的数目

xiaolai-sqlai commented 6 years ago

这是为了保持数目一致，因为我们最后一个batch只采751个id，每个id共计4张图片，故而要len和这个统一，不然在训练中，就会出现iter才跑了一半，整个epoch直接结束跳入下一个epoch的情况，主要是为了保证统一。