From 3dfad1e2d3ccb27955c75232b538f28753f2730b Mon Sep 17 00:00:00 2001 From: Xinyu Gong Date: Mon, 11 Nov 2019 11:55:04 -0600 Subject: [PATCH] Update function.py conduct flip operation with pytorch --- lib/core/function.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/lib/core/function.py b/lib/core/function.py index dadff0fa..1bc19daa 100755 --- a/lib/core/function.py +++ b/lib/core/function.py @@ -124,10 +124,7 @@ def validate(config, val_loader, val_dataset, model, criterion, output_dir, output = outputs if config.TEST.FLIP_TEST: - # this part is ugly, because pytorch has not supported negative index - # input_flipped = model(input[:, :, :, ::-1]) - input_flipped = np.flip(input.cpu().numpy(), 3).copy() - input_flipped = torch.from_numpy(input_flipped).cuda() + input_flipped = input.flip(3) outputs_flipped = model(input_flipped) if isinstance(outputs_flipped, list):