| #!/bin/bash |
| |
| readonly TEST_DATA="/usr/share/edgetpudemo" |
| readonly VIDEO_DEVICE_FILE="${TEST_DATA}/video_device.mp4" |
| readonly VIDEO_STREAM_FILE="${TEST_DATA}/video_stream.mp4" |
| readonly TPU_MODEL_FILE="${TEST_DATA}/mobilenet_ssd_v1_coco_quant_postprocess_edgetpu.tflite@Running MobileNet SSD v1 on Edge TPU" |
| readonly CPU_MODEL_FILE="${TEST_DATA}/mobilenet_ssd_v1_coco_quant_postprocess.tflite@Running MobileNet SSD v1 on CPU" |
| readonly LABELS_FILE="${TEST_DATA}/coco_labels.txt" |
| |
| if [[ "$1" == "--device" ]]; then |
| echo "Press 'q' to quit." |
| echo "Press 'n' to switch between models." |
| |
| edgetpu_detect \ |
| --source "${VIDEO_DEVICE_FILE}" \ |
| --model "${TPU_MODEL_FILE},${CPU_MODEL_FILE}" \ |
| --labels "${LABELS_FILE}" \ |
| --filter car,truck \ |
| --max_area 0.1 \ |
| --color white \ |
| --displaymode window |
| # Disable stream demo for now. |
| # elif [[ "$1" == "--stream" ]]; then |
| # echo "Press 'q' to quit." |
| # echo "Press 'n' to switch between models." |
| |
| # SERVER_INDEX_HTML="${TEST_DATA}/index.html" edgetpu_detect_server \ |
| # --source "${VIDEO_STREAM_FILE}" \ |
| # --model "${TPU_MODEL_FILE},${CPU_MODEL_FILE}" \ |
| # --labels "${LABELS_FILE}" \ |
| # --filter car,truck \ |
| # --max_area 0.1 \ |
| # --color white \ |
| # --loop |
| else |
| echo "Run on-device inference:" |
| echo " $0 --device" |
| # echo "Run streaming server:" |
| # echo " $0 --stream" |
| fi |