jkang commited on
Commit
809650c
1 Parent(s): 99f6dd9

upload readme

Browse files
.ipynb_checkpoints/model_card-checkpoint.md ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: MIT
3
+ language: en
4
+ license: mit
5
+ datasets:
6
+ - web crawled (coming soon)
7
+ ---
8
+
9
+ # Simple CNN-based Artist Classifier
10
+
11
+ This repo contains a simple CNN-based Keras model which classifies images into one of 10 selected artists/painters.
12
+
13
+ - The purpose of this model was for a quick prototyping
14
+ - Data has been web-crawled using `https://github.com/YoongiKim/AutoCrawler`
15
+ - 10 popular artists/painters were chosen:
16
+ - \[ARTIST\]: \[ID\]
17
+ - claude_monet: 0,
18
+ - henri_matisse: 1,
19
+ - jean_michel_basquiat: 2,
20
+ - keith_haring: 3,
21
+ - pablo_picasso: 4,
22
+ - pierre_augste_renoir: 5,
23
+ - rene_magritte: 6,
24
+ - roy_richtenstein: 7,
25
+ - vincent_van_gogh: 8,
26
+ - wassily_kandinsky: 9
27
+ - About 100 representative paintings per artist were crawled and manually checked
28
+ - Dataset will be shared later
29
+
30
+ # How to use
31
+
32
+ ```python
33
+ import tensorflow as tf
34
+ from huggingface_hub import from_pretrained_keras
35
+ model = from_pretrained_keras("jkang/drawing-artist-classifier")
36
+
37
+ image_file = 'cat.jpg'
38
+ img = tf.io.read_file(image_file)
39
+ img = tf.io.decode_jpeg(img, channels=3)
40
+
41
+ last_layer_activation, predictions = model(img[tf.newaxis,...])
42
+ ```
43
+
44
+ # Intended uses & limitations
45
+ You can use this model freely for predicting artists or trends of a given image.
46
+ Please keep in mind that this model is not intended for the production, but for a research and quick prototyping.
47
+ Web-crawled image data might not have balanced amount of drawings that sufficiently represent the artists.
48
+
49
+
50
+ ---
51
+ - 2022-01-18 first created by jaekoo kang
model_card.md ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: MIT
3
+ language: en
4
+ license: mit
5
+ datasets:
6
+ - web crawled (coming soon)
7
+ ---
8
+
9
+ # Simple CNN-based Artist Classifier
10
+
11
+ This repo contains a simple CNN-based Keras model which classifies images into one of 10 selected artists/painters.
12
+
13
+ - The purpose of this model was for a quick prototyping
14
+ - Data has been web-crawled using `https://github.com/YoongiKim/AutoCrawler`
15
+ - 10 popular artists/painters were chosen:
16
+ - \[ARTIST\]: \[ID\]
17
+ - claude_monet: 0,
18
+ - henri_matisse: 1,
19
+ - jean_michel_basquiat: 2,
20
+ - keith_haring: 3,
21
+ - pablo_picasso: 4,
22
+ - pierre_augste_renoir: 5,
23
+ - rene_magritte: 6,
24
+ - roy_richtenstein: 7,
25
+ - vincent_van_gogh: 8,
26
+ - wassily_kandinsky: 9
27
+ - About 100 representative paintings per artist were crawled and manually checked
28
+ - Dataset will be shared later
29
+
30
+ # How to use
31
+
32
+ ```python
33
+ import tensorflow as tf
34
+ from huggingface_hub import from_pretrained_keras
35
+ model = from_pretrained_keras("jkang/drawing-artist-classifier")
36
+
37
+ image_file = 'cat.jpg'
38
+ img = tf.io.read_file(image_file)
39
+ img = tf.io.decode_jpeg(img, channels=3)
40
+
41
+ last_layer_activation, predictions = model(img[tf.newaxis,...])
42
+ ```
43
+
44
+ # Intended uses & limitations
45
+ You can use this model freely for predicting artists or trends of a given image.
46
+ Please keep in mind that this model is not intended for the production, but for a research and quick prototyping.
47
+ Web-crawled image data might not have balanced amount of drawings that sufficiently represent the artists.
48
+
49
+
50
+ ---
51
+ - 2022-01-18 first created by jaekoo kang