kaprosdev commited on
Commit
82847d8
·
1 Parent(s): 37b422d

Upload model

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ variables/variables.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text
classes.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"not_photo": 0, "photo": 1}
example.py ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ # TF for image classification model
3
+
4
+ import tensorflow
5
+ import numpy
6
+ from PIL import Image
7
+
8
+ model = tensorflow.saved_model.load('./')
9
+ classes = [ "not_photo" , "photo" , ]
10
+
11
+ img = Image.open("image.jpg").convert('RGB')
12
+ img = img.resize((300, 300 * img.size[1] // img.size[0]), Image.ANTIALIAS)
13
+ inp_numpy = numpy.array(img)[None]
14
+
15
+
16
+ inp = tensorflow.constant(inp_numpy, dtype='float32')
17
+
18
+ class_scores = model(inp)[0].numpy()
19
+
20
+
21
+ print("")
22
+ print("class_scores", class_scores)
23
+ print("Class : ", classes[class_scores.argmax()])
saved_model.pb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7a525fa1db8c3c426c8e900c023c228f0f69743aac86749a29d0a79498820b77
3
+ size 4742794
variables/variables.data-00000-of-00001 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2e26d2f64d2b5f9f06320b64177af96f7ef44f88f8504018dd5c9868d073b49e
3
+ size 97526864
variables/variables.index ADDED
Binary file (16.9 kB). View file