Gs928
Gs928
for i, data in enumerate(testloader, 0): inputs = data.cuda() with torch.no_grad(): pos, height, offset = net(inputs) boxes = parse_det_offset(r, pos.cpu().numpy(), height.cpu().numpy(), offset.cpu().numpy(), config.size_test, score=0.1, down=4, nms_thresh=0.5) if len(boxes) > 0:...
print(input.shape)-->torch.Size([1, 3, 1024, 2048]) print(config.size_test)-->(1280, 2560) 我打断点也试过了，确实不一致
我想请教一下，您设计的初衷是想要让test阶段的size变为(1280, 2560)呢，还是就保持（1024, 2048）
嗯嗯，如果size保持1024 2048，指标略有变化，不过影响不大
(1280, 2560)这种尺寸我没试验过，但是保持1024 2048会使得结果略有改善。