From 4b3ff7a6775b22cdc02b8d51b45d81bbd83aeeb1 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Paul=20F=C3=A9ry?= Date: Thu, 29 Jun 2023 12:24:53 +0200 Subject: [PATCH] modify np.int -> int --- data/preprocessor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/data/preprocessor.py b/data/preprocessor.py index 25fad4279..a3d69e065 100755 --- a/data/preprocessor.py +++ b/data/preprocessor.py @@ -34,7 +34,7 @@ def __init__(self, data_root, seq_name, parser, log, split='train', phase='train assert False, 'error' self.gt = np.genfromtxt(label_path, delimiter=delimiter, dtype=str) - frames = self.gt[:, 0].astype(np.float32).astype(np.int) + frames = self.gt[:, 0].astype(np.float32).astype(int) fr_start, fr_end = frames.min(), frames.max() self.init_frame = fr_start self.num_fr = fr_end + 1 - fr_start @@ -88,7 +88,7 @@ def get_valid_id(self, pre_data, fut_data): return valid_id def get_pred_mask(self, cur_data, valid_id): - pred_mask = np.zeros(len(valid_id), dtype=np.int) + pred_mask = np.zeros(len(valid_id), dtype=int) for i, idx in enumerate(valid_id): pred_mask[i] = cur_data[cur_data[:, 1] == idx].squeeze()[-1] return pred_mask