Thundernet_Pytorch 多Batch输入

多Batch输入

Open 246yf87953246 opened this issue 5 years ago • 6 comments

请问一下，测试网络时支持多batchsize输入还是只能单batchsize？

Dec 26 '19 03:12 246yf87953246

可以多batch的，因为是输入shape固定为320 320

Dec 26 '19 04:12 DayBreak-u

我用的上一个版本的代码，因为上一个版本的代码我训练自己的数据结果较好；请问上一个版本的代码支持多batchsize吗？

Dec 26 '19 06:12 246yf87953246

上个版本不支持，除你自己padding到一个尺下，建议用新版本试试，新版的性能速度都更好

Dec 26 '19 06:12 DayBreak-u

新版本同样的数据集训练出来的MAP差有点多

Dec 26 '19 06:12 246yf87953246

好吧那可能有点问题，padding到同输入是可以的

Dec 26 '19 06:12 DayBreak-u

上个版本我已经统一尺度了，可以跑前向，但结果不对，都是背景

Dec 26 '19 06:12 246yf87953246