Skip to content

Commit

Permalink
Update to v2 of the mobilenet model, which seems more compatible in m…
Browse files Browse the repository at this point in the history
…ultiple environments (#194)

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

* Update readme

* Wording updates

* Spacing
  • Loading branch information
ladatz authored Sep 21, 2023
1 parent 1a806fc commit 12a7569
Show file tree
Hide file tree
Showing 3 changed files with 8 additions and 2 deletions.
Original file line number Diff line number Diff line change
@@ -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.
Git LFS file not shown
Original file line number Diff line number Diff line change
Expand Up @@ -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();

Expand Down

0 comments on commit 12a7569

Please sign in to comment.