Welcome to the DiffuseGaitNet repository! This project implements the research paper titled "DiffuseGaitNet: Improving Parkinson’s Disease Gait Severity Assessment with Diffusion Model Framework". The project aims to improve the assessment of gait severity in Parkinson's Disease using a novel diffusion model framework.
This implementation allows users to experiment with the DiffuseGaitNet model, although the original dataset used in the paper is not publicly available. You can, however, run the model using your own data if structured similarly.
- Python 3.10: Make sure you have Python 3.10 installed. You can download it from python.org.
-
Clone the Repository
Start by cloning the repository to your local machine:
git clone https://github.com/arshakRz/DiffuseGaitNet.git cd DiffuseGaitNet -
Create and Activate the Virtual Environment
Create a virtual environment to isolate project dependencies:
python3 -m venv venv
Activate the virtual environment:
-
On macOS and Linux:
source venv/bin/activate -
On Windows:
venv\Scripts\activate
-
-
Install Dependencies
With the virtual environment activated, install the required dependencies:
pip install -r requirements.txt
To generate sample animations from a model checkpoint, you can use animate.py:
python utils/animate.pyMake sure to set up the appropriate directories and paths in the config.py file before running the script. This will allow you to generate sample outputs based on your own data.
Before running any scripts, configure the necessary directories and paths in config.py according to your file structure and data paths.
We hope you find DiffuseGaitNet useful for your research or personal projects! If you encounter any issues or have questions, please open an issue in the repository.