From 12a75699988db62ac7d416dad5460939cfad4aa6 Mon Sep 17 00:00:00 2001
From: ladatz <105828115+ladatz@users.noreply.github.com>
Date: Thu, 21 Sep 2023 13:12:56 -0400
Subject: [PATCH] Update to v2 of the mobilenet model, which seems more
 compatible in multiple environments (#194)

* Update to v2 of the mobilenet model, which seems more compatible in multiple environments

* Update readme

* Wording updates

* Spacing
---
 .../applications/local-object-detection/models/readme.txt    | 5 ++++-
 .../local-object-detection/models/ssd_mobilenet_v2_coco.pb   | 3 +++
 .../applications/local-object-detection/src/detection.rs     | 2 +-
 3 files changed, 8 insertions(+), 2 deletions(-)
 create mode 100644 examples/applications/local-object-detection/models/ssd_mobilenet_v2_coco.pb

diff --git a/examples/applications/local-object-detection/models/readme.txt b/examples/applications/local-object-detection/models/readme.txt
index 0ecd1a20..15eb3f4b 100644
--- a/examples/applications/local-object-detection/models/readme.txt
+++ b/examples/applications/local-object-detection/models/readme.txt
@@ -1 +1,4 @@
-Model used in this folder is 'ssd_mobilenet_v1_coco' taken from https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/tf1_detection_zoo.md
+Model used in this folder is 'ssd_mobilenet_v2_coco' taken from
+https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/tf1_detection_zoo.md.
+This folder also contains 'ssd_mobilenet_v1_coco' from the same directory, which can be used in WSL
+Ubuntu 20.04 but has had some compatibility issues in other environments.
diff --git a/examples/applications/local-object-detection/models/ssd_mobilenet_v2_coco.pb b/examples/applications/local-object-detection/models/ssd_mobilenet_v2_coco.pb
new file mode 100644
index 00000000..0b2a1bac
--- /dev/null
+++ b/examples/applications/local-object-detection/models/ssd_mobilenet_v2_coco.pb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2a8d8a89d695842e60d8c6d144181100555563e21acf2fa1e8f561fec5c3c6ad
+size 69688296
diff --git a/examples/applications/local-object-detection/src/detection.rs b/examples/applications/local-object-detection/src/detection.rs
index 938cbbcf..11e3cae8 100644
--- a/examples/applications/local-object-detection/src/detection.rs
+++ b/examples/applications/local-object-detection/src/detection.rs
@@ -25,7 +25,7 @@ pub struct DetectionLogic {
 impl DetectionLogic {
     pub fn new() -> Self {
         let mut graph = Graph::new();
-        let proto = include_bytes!("../models/ssd_mobilenet_v1_coco.pb");
+        let proto = include_bytes!("../models/ssd_mobilenet_v2_coco.pb");
 
         graph.import_graph_def(proto, &ImportGraphDefOptions::new()).unwrap();