diff --git a/README.md b/README.md index 0a5a27a..fd4380d 100644 --- a/README.md +++ b/README.md @@ -8,7 +8,7 @@ together with the Edge TPU Python API. Before you start using the examples run the ```download_models.sh``` script in order to download a selection of models. These canned models will be downloaded and extracted to a new folder -```canned_models```. +```all_models```. Further requirements may be needed by the different camera libraries, check the diff --git a/download_models.sh b/download_models.sh index 471b3e6..7a9dac5 100644 --- a/download_models.sh +++ b/download_models.sh @@ -13,4 +13,7 @@ # See the License for the specific language governing permissions and # limitations under the License. -wget http://storage.googleapis.com/cloud-iot-edge-pretrained-models/canned_models.tar.gz +mkdir -p all_models +wget https://dl.google.com/coral/canned_models/all_models.tar.gz +tar -C all_models -xvzf all_models.tar.gz +rm -f all_models.tar.gz diff --git a/gstreamer/classify.py b/gstreamer/classify.py index f2d905f..826a6d2 100644 --- a/gstreamer/classify.py +++ b/gstreamer/classify.py @@ -34,7 +34,7 @@ def generate_svg(dwg, text_lines): dwg.add(dwg.text(line, insert=(10, y*20), fill='white', font_size='20')) def main(): - default_model_dir = "../canned_models" + default_model_dir = "../all_models" default_model = 'mobilenet_v2_1.0_224_quant_edgetpu.tflite' default_labels = 'imagenet_labels.txt' parser = argparse.ArgumentParser() diff --git a/gstreamer/detect.py b/gstreamer/detect.py index 54dcb4d..c803ddc 100644 --- a/gstreamer/detect.py +++ b/gstreamer/detect.py @@ -59,7 +59,7 @@ def generate_svg(dwg, objs, labels, text_lines): fill='red', fill_opacity=0.3, stroke='white')) def main(): - default_model_dir = '../canned_models' + default_model_dir = '../all_models' default_model = 'mobilenet_ssd_v2_coco_quant_postprocess_edgetpu.tflite' default_labels = 'coco_labels.txt' parser = argparse.ArgumentParser() diff --git a/raspicam/README.md b/raspicam/README.md index 5a011bd..8be7113 100644 --- a/raspicam/README.md +++ b/raspicam/README.md @@ -24,6 +24,6 @@ python3 classify_capture.py You can change the model and the labels file using flags: ``` -python3 classify_capture.py --model ../canned_models/inception_v3_299_quant_edgetpu.tflite +python3 classify_capture.py --model ../all_models/inception_v3_299_quant_edgetpu.tflite ``` diff --git a/raspicam/classify_capture.py b/raspicam/classify_capture.py index 5bb8757..df12b73 100644 --- a/raspicam/classify_capture.py +++ b/raspicam/classify_capture.py @@ -23,7 +23,7 @@ import edgetpu.classification.engine def main(): - default_model_dir = "../canned_models" + default_model_dir = "../all_models" default_model = 'mobilenet_v2_1.0_224_quant_edgetpu.tflite' default_labels = 'imagenet_labels.txt' parser = argparse.ArgumentParser()