A deep learning method for multimodal pain assessment with transformers.
Clone the repository and follow the instructions below to set up the environment and start training.
git clone https://github.com/your-org/PainFusion.git
cd PainFusionA Dockerfile is provided to create a reproducible environment.
# Build the image (run from repo root)
docker build -t painfusion:latest .
# Run a container interactively (adjust flags as needed)
docker run --rm -it painfusion:latest /bin/bashAll configuration files live in the configs/ directory. An example configuration is included as configs/default.yaml to demonstrate expected structure and defaults.
When launching scripts, point them to the desired YAML configuration file.
To start training, run the provided script and pass the config file path as an argument:
bash scripts/run-train.sh configs/config_file.yamlPainFusion+/
├── configs/ # Configuration files (example included: configs/example.yaml)
├── scripts/ # Training and utility scripts
├── data/ # Data loading and preprocessing
├── modules/ # Model components
├── docker # Docker and Python dependencies
├── README.md # This file
├── main.py # Main training script
└── LICENSE # License
If you use PainFusion+ in your research or work, please cite our paper (citation to be added once published):
Citation — TO DO (will be provided once the work is published)
For questions, suggestions, or other correspondence, please open a pull request or send an email to: mbenavent@dtic.ua.es.