search

daizuozhuo / batch-dropblock-network

Official source code of "Batch DropBlock Network for Person Re-identification and Beyond" (ICCV 2019)
MIT License
326 stars 90 forks source link

test #13

Open chj1933 opened 5 years ago

chj1933 commented 5 years ago

when I test, there is a error, ValueError: expected 2D or 3D input (got 1D input), please tell me how should I do.

wzjiang commented 5 years ago

我也遇到一样的问题，在market1501，cuhk不会，在clothes就会

chj1933 commented 5 years ago

@wzjiang ,请问你是怎样处理 ValueError: expected 2D or 3D input (got 1D input)？

liqb-a commented 5 years ago

出现这个的原因是 batch为1（测试batch_size为1 或者 最后一批图片个数为1）；根本原因是因为作者在搭建网络的时候有做压缩处理，维度大小为1的都会被压缩。 修改： 1、修改网络，压缩增加判断，如果维度为1在扩维。 2、更改batch_size和测试图片个数（让batch不为1即可）

YunhaoDu commented 4 years ago

@liqb-a 经过测试，发现batch_size<=3就会报这个错误；请问源码中是在哪里进行的压缩处理呢？怎么去修改呢？

yukaizhou commented 4 years ago

我也遇到了这个问题，是因为网络（network.py）中使用.squeeze进行维度压缩，导致在batchsize为1时，batchsize的维度也没干掉了，使用view改写下就可以了。使用给的模型结果是85.5和94.6%。