test

[chj1933]

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

[wzjiang]

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

[chj1933]

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

[liqb-a]

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

[YunhaoDu]

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

[yukaizhou]

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