batch-dropblock-network icon indicating copy to clipboard operation
batch-dropblock-network copied to clipboard

test

Open chj1933 opened this issue 5 years ago • 5 comments

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

chj1933 avatar Jul 18 '19 06:07 chj1933

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

wzjiang avatar Jul 19 '19 08:07 wzjiang

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

chj1933 avatar Jul 30 '19 02:07 chj1933

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

liqb-a avatar Aug 14 '19 07:08 liqb-a

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

YunhaoDu avatar Mar 05 '20 13:03 YunhaoDu

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

yukaizhou avatar Jul 30 '20 09:07 yukaizhou