batch-dropblock-network
batch-dropblock-network copied to clipboard
test
when I test, there is a error, ValueError: expected 2D or 3D input (got 1D input), please tell me how should I do.
我也遇到一样的问题,在market1501,cuhk不会,在clothes就会
@wzjiang ,请问你是怎样处理 ValueError: expected 2D or 3D input (got 1D input)?
出现这个的原因是 batch为1(测试batch_size为1 或者 最后一批图片个数为1);根本原因是因为作者在搭建网络的时候有做压缩处理,维度大小为1的都会被压缩。 修改: 1、修改网络,压缩增加判断,如果维度为1在扩维。 2、更改batch_size和测试图片个数(让batch不为1即可)
@liqb-a 经过测试,发现batch_size<=3就会报这个错误;请问源码中是在哪里进行的压缩处理呢?怎么去修改呢?
我也遇到了这个问题,是因为网络(network.py)中使用.squeeze进行维度压缩,导致在batchsize为1时,batchsize的维度也没干掉了,使用view改写下就可以了。使用给的模型结果是85.5和94.6%。