| #!/bin/bash |
| |
| # Prereq: |
| # g4d -f NAME |
| # blaze build third_party/tensorflow/lite/tools:visualize |
| |
| ANDROID_BUILD_TOP="/usr/local/google/home/$(whoami)/android/master" |
| MODEL_DIR="$ANDROID_BUILD_TOP/test/mlts/models/assets" |
| # The .json files are always output to /tmp |
| HTML_DIR="/tmp" |
| |
| mkdir -p $HTML_DIR |
| |
| for file in "$MODEL_DIR"/*.tflite |
| do |
| if [ -f "$file" ]; then |
| filename=`basename $file` |
| modelname=${filename%.*} |
| blaze-bin/third_party/tensorflow/lite/tools/visualize $file $HTML_DIR/$modelname.html |
| fi |
| done |
| |
| # Example visualization: blaze-bin/third_party/tensorflow/lite/tools/visualize ~/android/master/test/mlts/models/assets/mobilenet_v1_0.75_192.tflite /tmp/mobilenet_v1_0.75_192.html |