Create README.md

README.md

@@ -0,0 +1,47 @@
+---
+language: en
+license: mit
+datasets:
+- web crawled (coming soon)
+---
+
+# Simple CNN-based Artist Classifier
+
+This repo contains a simple CNN-based Keras model which classifies images into one of 10 selected artists/painters.
+
+- The purpose of this model was for a quick prototyping
+- Data has been web-crawled using `https://github.com/YoongiKim/AutoCrawler`
+- 10 popular artists/painters were chosen:
+    - \[ARTIST\]: \[ID\]
+    - claude_monet: 0,
+    - henri_matisse: 1,
+    - jean_michel_basquiat: 2,
+    - keith_haring: 3,
+    - pablo_picasso: 4,
+    - pierre_augste_renoir: 5,
+    - rene_magritte: 6,
+    - roy_richtenstein: 7,
+    - vincent_van_gogh: 8,
+    - wassily_kandinsky: 9
+- About 100 representative paintings per artist were crawled and manually checked
+- Dataset will be shared later
+
+# How to use
+```python
+import tensorflow as tf
+from huggingface_hub import from_pretrained_keras
+model = from_pretrained_keras("jkang/drawing-artist-classifier")
+
+image_file = 'cat.jpg'
+img = tf.io.read_file(image_file)
+img = tf.io.decode_jpeg(img, channels=3)
+
+last_layer_activation, predictions = model(img[tf.newaxis,...])
+```
+
+# Intended uses & limitations
+You can use this model freely for predicting artists or trends of a given image.
+Please keep in mind that this model is not intended for production, but for research and quick prototyping.
+Web-crawled image data might not have a balanced amount of drawings that sufficiently represent the artists.
+---
+- 2022-01-18 first created by jaekoo kang