我用了demo.py 提示维度问题

yqyao / SSD_Pytorch

support different SSDs and different scale test, support refineDet.

MIT License

148 stars 51 forks source link

我用了demo.py 提示维度问题 #26

Closed luvWY closed 5 years ago

luvWY commented 5 years ago

我采用了 refine_res101_voc-.yaml的配置文件然后在训练时候保存的refine_res_epoch_210_300.pth 我eval是可以通过的，但是在demo.py时 在im_dete的函数里面output = net(x)报错： The size of tensor a (19) must match the size of tensor b (20) at non-singleton dimension 3`

我不是很懂，我的模型和load的参数应该是匹配的，为什么会报错呢。请教一下！

luvWY commented 5 years ago

好像是config里面的参数问题 INPUT_WH:设置成 320就可以跑通但是和训练一样的300大小就不行为什么

yqyao commented 5 years ago

refineDet 只能支持320的输入，64的倍数，因为那个反卷积的关系 @luvWY

luvWY commented 5 years ago

refineDet 只能支持320的输入，64的倍数，因为那个反卷积的关系 @luvWY

了解了

rw1995 commented 5 years ago

refineDet 只能支持320的输入，64的倍数，因为那个反卷积的关系 @luvWY

了解了

请问下如何合并数据集，我已经困扰了2天了，求助 @luvWY ，main文件夹下的txt实在太多了