Model Upload

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+variables/variables.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text

classes.json

@@ -0,0 +1 @@
+{"Safe": 0, "Naked": 1}

example.py

@@ -0,0 +1,23 @@
+
+# TF for image classification model
+
+import tensorflow
+import numpy
+from PIL import Image
+
+model = tensorflow.saved_model.load('./')
+classes = [ "Safe" ,  "Naked" , ]
+
+img = Image.open("image.jpg").convert('RGB')
+img = img.resize((300, 300 * img.size[1] // img.size[0]), Image.ANTIALIAS)
+inp_numpy = numpy.array(img)[None]
+
+
+inp = tensorflow.constant(inp_numpy, dtype='float32')
+
+class_scores = model(inp)[0].numpy()
+
+
+print("")
+print("class_scores", class_scores)
+print("Class : ", classes[class_scores.argmax()])

saved_model.pb

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ab9e15060ca77171a860cbf23130ca5745bda2de895ad99586fb0b52df413cbb
+size 6355003

variables/variables.data-00000-of-00001

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:07caa8236fd4fddc7581ef6286e248a8a108a555d7e327ec37638efedc3450fc
+size 16992324