Awesome
3D Hand Pose Estimation from Single RGB Camera
Master's Thesis in Ukrainian Catholic University (2018)
All the details on the data, preprocessing, model architecture and training details can be found in thesis text.
Experiment weights can be downloaded from Google Drive.
Requirements
Requirements are specified in requirements.txt
.
pip install -r requirements.txt
Model works only on cuda.
Usage
There are two main scripts - trainer.py
and evaluate.py
, which are used for training and evaluation.
python trainer.py {experiment_name}
python evaluate.py {experiment_name} ({dataset_name})
All parameters are specified in config.yaml
file in corresponding experiment folder.
Examples:
python trainer.py e010
python evaluate.py e010 dexter+object
Project structure
Folders dataset
, model
, criterion
, metric
, optimizer
contain datasets, models, losses, metrics for evaluation and optimizers, respectively.
To get a specific dataset, model, loss, metric or optimizer, call functions get_dataloder
, get_model
, get_criterion
, get_metric
or get_optimizer
, respectively. Functions are defined in __init__.py
files in corresponding folders.