Height & Weight Finder
This is a Python script to find height, weight, segmentation mask and joint locations of a person from a full-body single person image. Used networks are implemented in "Height and Weight Estimation From Unconstrained Images" paper. You can find this paper's repository here.
Reference
If you find this program useful in your research, please consider citing:
@inproceedings{altinigne2020height,
title={Height and Weight Estimation from Unconstrained Images},
author={Altinigne, Can Yilmaz and Thanou, Dorina and Achanta, Radhakrishna},
booktitle={ICASSP 2020-2020 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP)},
pages={2298--2302},
year={2020},
organization={IEEE}
}
Trained Models in the Original Paper
Each file is nearly 2GB. Just download them and put them in /models
folder.
- Pretrained Height Network: https://drive.google.com/open?id=1fX0DDgbTcOOmiz9KdtU7I2YYg5S49upj
- Pretrained Weight Network: https://drive.google.com/open?id=14ShT0rsUohiGT0wJlKY9cGHgEy0w4Ity
Dependency
- PyTorch = 1.2.0
- Python = 3.6
Usage
In order to get outputs, just run python HWFinder.py -i [IMAGE ADDRESS] -g [GPU NUMBER] -r [RESOLUTION]
. The models are trained using images with a resolution of 128x128
, so set the RESOLUTION
parameter between 128 and 256.
Environment
You can use environment.yml
file to create a Conda environment to run the script. You can create a new environment using this command. conda env create -f environment.yml -p [PATH]