crosstyan/OpenGait
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update pretreatment_SUSTech1K.py

Browse Source
This commit is contained in:
SHEN CHUANFU
2025-09-24 21:18:51 +08:00
committed by GitHub
parent 74883d8373
commit ae87f04d62
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
datasets/SUSTech1K/pretreatment_SUSTech1K.py
+2 -2
View File
@@ -126,8 +126,8 @@ def imgs2pickle(img_groups: Tuple, output_path: Path, img_size: int = 64, verbos
HWs = [img.shape[:2] for img in imgs] HWs = [img.shape[:2] for img in imgs]
# transpose to (C, H W) # transpose to (C, H W)
data = [cv2.resize(img, (img_size, img_size), interpolation=cv2.INTER_CUBIC) for img in imgs] data = [cv2.resize(img, (img_size, img_size), interpolation=cv2.INTER_CUBIC) for img in imgs]
imgs = [img.transpose(2, 0, 1) for img in imgs] imgs = [img.transpose(2, 0, 1) for img in data]
data = np.asarray(data) data = np.asarray(imgs)
HWs = np.asarray(HWs) HWs = np.asarray(HWs)
elif modality == 'Sils_raw': elif modality == 'Sils_raw':
sils = [cv2.imread(sil, cv2.IMREAD_GRAYSCALE) for sil in data_files] sils = [cv2.imread(sil, cv2.IMREAD_GRAYSCALE) for sil in data_files]