blob: 5962c905047fff6aa55367b5853764232c16081d (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
|
From 17e03724c925f8d0b11c8f14699646aa498af798 Mon Sep 17 00:00:00 2001
From: Hongxu Jia <hongxu.jia@windriver.com>
Date: Mon, 16 May 2022 00:36:00 -0700
Subject: [PATCH] label_image: tweak default model location
Upstream-Status: Inappropriate [oe specific]
Signed-off-by: Hongxu Jia <hongxu.jia@windriver.com>
---
tensorflow/examples/label_image/label_image.py | 6 +++---
tensorflow/examples/label_image/main.cc | 6 +++---
2 files changed, 6 insertions(+), 6 deletions(-)
diff --git a/tensorflow/examples/label_image/label_image.py b/tensorflow/examples/label_image/label_image.py
index 3e4bc1ddaab..83558251d77 100644
--- a/tensorflow/examples/label_image/label_image.py
+++ b/tensorflow/examples/label_image/label_image.py
@@ -62,10 +62,10 @@ def load_labels(label_file):
if __name__ == "__main__":
- file_name = "tensorflow/examples/label_image/data/grace_hopper.jpg"
+ file_name = "/usr/share/label_image/grace_hopper.jpg"
model_file = \
- "tensorflow/examples/label_image/data/inception_v3_2016_08_28_frozen.pb"
- label_file = "tensorflow/examples/label_image/data/imagenet_slim_labels.txt"
+ "/usr/share/label_image/inception_v3_2016_08_28_frozen.pb"
+ label_file = "/usr/share/label_image/imagenet_slim_labels.txt"
input_height = 299
input_width = 299
input_mean = 0
diff --git a/tensorflow/examples/label_image/main.cc b/tensorflow/examples/label_image/main.cc
index 13bb5980276..38935a57dd4 100644
--- a/tensorflow/examples/label_image/main.cc
+++ b/tensorflow/examples/label_image/main.cc
@@ -279,11 +279,11 @@ int main(int argc, char* argv[]) {
// They define where the graph and input data is located, and what kind of
// input the model expects. If you train your own model, or use something
// other than inception_v3, then you'll need to update these.
- string image = "tensorflow/examples/label_image/data/grace_hopper.jpg";
+ string image = "/usr/share/label_image/grace_hopper.jpg";
string graph =
- "tensorflow/examples/label_image/data/inception_v3_2016_08_28_frozen.pb";
+ "/usr/share/label_image/inception_v3_2016_08_28_frozen.pb";
string labels =
- "tensorflow/examples/label_image/data/imagenet_slim_labels.txt";
+ "/usr/share/label_image/imagenet_slim_labels.txt";
int32_t input_width = 299;
int32_t input_height = 299;
float input_mean = 0;
--
2.36.0
|
