From 8cd55a2095df3370392bfdb8d7fa8461b29cd78d Mon Sep 17 00:00:00 2001 From: Xlxw Date: Tue, 7 Mar 2023 15:00:58 +0800 Subject: [PATCH] fix bug in data_utils.py for Issue #10 --- model/STG-correction/utils/data_utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/model/STG-correction/utils/data_utils.py b/model/STG-correction/utils/data_utils.py index 9674e34..c13c0b4 100644 --- a/model/STG-correction/utils/data_utils.py +++ b/model/STG-correction/utils/data_utils.py @@ -425,13 +425,15 @@ def convert_spmap_sw(sp_map : list, sw_labels : np.array) -> list: def convert_spmap_tg(sp_map : list, tg_mapper : list) -> list: new_spmap = [] for idx in range(len(sp_map)): + if idx == 163: + print() spmapper = sp_map[idx] tgmapper = tg_mapper[idx] mapper = {} if not spmapper: new_spmap.append(None) else: for sp in spmapper: - if sp in tgmapper and tgmapper[sp] > 0: mapper[tgmapper[sp]] = spmapper[sp] + if sp in tgmapper and tgmapper[sp] >= 0: mapper[tgmapper[sp]] = spmapper[sp] new_spmap.append(mapper) return new_spmap