Update the model to v2 (#1)

dog-ceo · Nov 26, 2020 · b04b9c1 · b04b9c1
1 parent 2114e78
commit b04b9c1
Show file tree

Hide file tree

Showing 10 changed files with 8 additions and 4 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -1,5 +1,7 @@
 FROM tensorflow/tensorflow:latest
 
+ARG VERSION
+
 WORKDIR /train
 
 COPY ./images /train/images
@@ -11,13 +13,13 @@ RUN make_image_classifier \
   --image_dir /train/images \
   --tfhub_module https://tfhub.dev/google/tf2-preview/mobilenet_v2/feature_vector/4 \
   --image_size 224 \
-  --saved_model_dir data/model/1 \
+  --saved_model_dir data/model/$VERSION \
   --labels_output_file data/class_labels.txt \
   --train_epochs 15 \
   --do_fine_tuning
 
 RUN tensorflowjs_converter \
   --input_format=tf_saved_model \
   --output_format=tfjs_graph_model \
-  data/model/1 \
-  web/model/1
+  data/model/$VERSION \
+  web/model/$VERSION
diff --git a/data/class_labels.txt b/data/class_labels.txt
@@ -1,3 +1,4 @@
+.git
 affenpinscher
 african
 airedale

diff --git a/data/model/2/saved_model.pb b/data/model/2/saved_model.pb
diff --git a/data/model/2/variables/variables.data-00000-of-00001 b/data/model/2/variables/variables.data-00000-of-00001
diff --git a/data/model/2/variables/variables.index b/data/model/2/variables/variables.index
diff --git a/public/model/2/group1-shard1of3.bin b/public/model/2/group1-shard1of3.bin
diff --git a/public/model/2/group1-shard2of3.bin b/public/model/2/group1-shard2of3.bin
diff --git a/public/model/2/group1-shard3of3.bin b/public/model/2/group1-shard3of3.bin
diff --git a/public/model/2/model.json b/public/model/2/model.json
diff --git a/src/App.tsx b/src/App.tsx
@@ -12,7 +12,7 @@ function App() {
 
   useEffect(() => {
     (async () => {
-      setModel(await tf.loadGraphModel('model/1/model.json'));
+      setModel(await tf.loadGraphModel('model/2/model.json'));
     })();
   }, []);