Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
models
test_set_images
README.md
features.py
image_util.py
model_wrapper.py
predict_mass.py
predictions.txt
produce_latex_chart.py
thickness_model.h5
util.py

README.md

image2mass

Code for http://proceedings.mlr.press/v78/standley17a.html

setup (works only with python3):

Install tensorflow

Download evaluation datafiles (see below)

use pip

pip install Pillow
pip install keras==2.1.1 # the most recent version (2.1.2 at time of writing) throws an error loading the model
pip install lz4
pip install opencv-python # or install opencv another way (this method doesn't support cv.imshow() or FFmpeg) but is much easier to install
pip install h5py

conda

Coming soon

example usage:

python3 predict_mass.py test_set_images/airplane_clock_1.jpg 6.25 1.25 2.125 2>/dev/null

(more in predictions.txt)

Ground Truth for Household Test Set:

https://docs.google.com/spreadsheets/d/1-cjqxaG8AGP14KvUZxSQzLGfGmS3_IbVO47_vbBKyoc

Evaluation Data Files:

https://drive.google.com/drive/folders/17yEukxIyjyen3vJ8nuX_YklnsVBZBol4

Amazon train/val/test Data Files

https://goo.gl/forms/hCLNHOcv7NRkEQP03