From 36a4faff4eb353052fe7f554034e937363a45387 Mon Sep 17 00:00:00 2001 From: unanmed <1319491857@qq.com> Date: Tue, 31 Mar 2026 22:35:28 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E6=8A=A5=E9=94=99?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ginka/dataset.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/ginka/dataset.py b/ginka/dataset.py index 5558942..b50d463 100644 --- a/ginka/dataset.py +++ b/ginka/dataset.py @@ -35,17 +35,20 @@ class GinkaMaskGITDataset(Dataset): if np.random.rand() > 0.5: k = np.random.randint(0, 4) target_np = np.rot90(target_np, k) - heatmap = np.rot90(heatmap, k) + for i in range(0, heatmap.shape[0]): + heatmap[i] = np.rot90(heatmap[i], k) if np.random.rand() > 0.5: target_np = np.fliplr(target_np) - heatmap = np.fliplr(heatmap) + for i in range(0, heatmap.shape[0]): + heatmap[i] = np.fliplr(heatmap[i]) if np.random.rand() > 0.5: target_np = np.flipud(target_np) - heatmap = np.flipud(heatmap) + for i in range(0, heatmap.shape[0]): + heatmap[i] = np.flipud(heatmap[i]) - target = torch.LongTensor(target_np) # [H, W] + target = torch.LongTensor(target_np.copy()) # [H, W] cond = torch.FloatTensor(item['val']) # [cond_dim] if random.random() < 0.5: