Update README.md

README.md

@@ -1,3 +1,26 @@
----
-license: mit
----
+---
+license: mit
+datasets:
+- google/quickdraw
+pipeline_tag: image-feature-extraction
+---
+
+A simple, small-ish network for producing embeddings for black and white binary images.  Takes a 32x32 drawing a produces a 64-dimensional embedding.
+
+Sample usage:
+
+```
+import onnxruntime as ort
+import numpy
+
+ort_sess = ort.InferenceSession('tiny_doodle_embedding.onnx')
+
+def compare(input_img_a, input_img_b):
+    img_a = process_input(input_img_a)  # Crop and resize the input image so it's binary and fits in a 32x32 array.
+    img_b = process_input(input_img_b)
+
+    a_embedding = ort_sess.run(None, {'input': img_a.astype(numpy.float32)})[0]
+    b_embedding = ort_sess.run(None, {'input': img_b.astype(numpy.float32)})[0]
+
+    sim = numpy.dot(a_embedding , b_embedding.T)  # Or a_embedding @ b_embedding.T
+```