parkinsons_Motor / downsample.py
virustechhacks's picture
force sync local parkinsons_Motor
d67c79d verified
import os
from pathlib import Path
def shrink(filename, skip_factor=100):
p = Path(filename)
if not p.exists(): return
print(f"Shrinking {p.name} from {p.stat().st_size / 1e6:.2f} MB...")
with open(p, 'r') as f:
lines = f.readlines()
downsampled = lines[::skip_factor]
with open(p, 'w') as f:
f.writelines(downsampled)
print(f"Done! New size: {p.stat().st_size / 1e6:.2f} MB.")
if __name__ == "__main__":
results_dir = Path("fleming-model-based-brain/Model_Results")
shrink(results_dir / "Force_amplitude_values.csv")
shrink(results_dir / "Force_times.csv")
shrink(results_dir / "sEMG_values.csv")
shrink(results_dir / "sEMG_times.csv")
print("Deleting unused .mat files...")
import glob
for mat_file in glob.glob(str(results_dir / "*.mat")):
os.remove(mat_file)
print("Removed", mat_file)