idealo / image-quality-assessment

Convolutional Neural Networks to predict the aesthetic and technical quality of images.

Home Page:https://idealo.github.io/image-quality-assessment/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to save output of batch prediction?

mmz15 opened this issue · comments

commented

I've looked at https://github.com/idealo/image-quality-assessment/issues/72 but found that I didn't contain enough information to answer my question.

I'm using the batch prediction functionality with the predict executable. My question: How do I save the json output? I've tried:

./predict  \
--docker-image nima-cpu \
--base-model-name MobileNet \
--weights-file $(pwd)/models/MobileNet/weights_mobilenet_technical_0.11.hdf5 \
--image-source $(pwd)/src/tests/test_images
--output $(pwd)/src/out.json

But it doesn't work. I've tried variations of output as well. What's the correct flag?