Skip to content

Latest commit

 

History

History
 
 

visualization

README.md

Visualization tools

This directory contains some stand-alone Python utility scripts that you can use to visualize various incoming and predicted labels on images.

Environment setup

Please see the Installation section in the main README for instructions on setting up the environment to run these scripts in.

Visualize detector output

visualize_detector_output.py draws the bounding boxes, their confidence level and predicted category annotated on top of the original images, and saves the annotated images to another directory. The original images can be in a local directory or in Azure Blob Storage.

Please see the top of visualize_detector_output.py for the arguments it requires.

  • If you are not running this on the computer with the original images, the script can download them from Azure Blob Storage using a SAS key to the container (supplied as the --sas_url argument, please surround the SAS URL by double quotes). It takes about 1.5 seconds per image, depending on your location and network speed. The SAS key looks like
https://storageaccountname.blob.core.windows.net/container-name?se=2019-04-06T23%3A38%3A00Z&sp=rl&sv=2018-03-28&sr=c&sig=A_LONG_STRING
  • You can choose to render a sample of n images by supplying the --sample argument.

Example invocations

It is best to call the script from the root dir of this repo so the path to the repo is on the PYTHONPATH.

Example invocation of the script, images stored locally:

python visualization/visualize_detector_output.py path_to/requestID_detections.json rendered_images_dir --confidence 0.9 --images_dir path_to_root_dir_of_original_images

Another example, for images stored in Azure Blob Storage and drawing a sample of 20 images:

python visualize_detector_output.py path_to/requestID_detections.json rendered_images_dir --confidence 0.9 --sas_url "https://storageaccountname.blob.core.windows.net/container-name?se=2019-04-06T23%3A38%3A00Z&sp=rl&sv=2018-03-28&sr=c&sig=A_LONG_STRING" --sample 20

If you encounter an error where it complains about not finding the module visualization_utils, you need to append the absolute path to the current directory to your PYTHONPATH. At your terminal or command line:

export PYTHONPATH=$PYTHONPATH:/absolute_path/CameraTraps