zc/mlp-project
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

TEST: use tensorboard for stuff

Browse source
This commit is contained in:
Zhengyi Chen 2024-03-04 20:32:26 +00:00
parent 2d31162c58
commit 524ee03187
10 changed files with 212 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

4
preprocess_data.py
View file

@ -25,6 +25,7 @@ import cv2
import scipy.io as io
import scipy.sparse as sparse
import h5py
from tqdm.auto import tqdm
CWD = os.getcwd()
@ -51,7 +52,7 @@ def pre_dataset_sh():
# np.random.seed(0)
# random.seed(0)
for img_path in img_paths:
for _, img_path in tqdm(img_paths, desc="Preprocessing Data"):
img_data = cv2.imread(img_path)
mat = io.loadmat(
img_path
@ -169,4 +170,5 @@ def make_npydata():
if __name__ == "__main__":
# Download manually...
pre_dataset_sh() # XXX: preliminary
print("Storing dataset paths...")
make_npydata()