tiny_quick_draw_recognition/config.yaml at main · mapapin/tiny_quick_draw_recognition · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
image_size:
  - 28
  - 28

loss_name: cross_entropy

epochs: 15
learning_rate: 0.0001

wandb_parameters:
  project_name: Tiny Quick Draw
  username: mapapin

data:
  folder: dataset
  generate: True
  train_samples: 3000
  validation_samples: 500
  test_samples: 500
  batch_size: 64
  classes:
    - airplane
    - ant
    - apple
    - arm
    - banana
    - basket
    - basketball
    - bat
    - bathtub
    - beach
    - bed
    - bee
    - bicycle
    - bird
    - birthday cake
    - book
    - bowtie
    - bridge
    - broom
    - bucket
    - bus
    - butterfly
    - cactus
    - camera
    - campfire
    - candle
    - car
    - carrot
    - cell phone
    - chair
    - circle
    - clock
    - cloud
    - computer
    - cookie
    - couch
    - crown
    - cup
    - diamond
    - donut
    - door
    - ear
    - envelope
    - eye
    - eyeglasses
    - face
    - fish
    - flower
    - guitar
    - hamburger
    - hand
    - key
    - ladder
    - moon
    - panda
    - pants
    - pear
    - pizza
    - sailboat
    - skull
    - snowflake
    - strawberry
    - television
    - tooth
    - tree
    - triangle
    - umbrella
    - whale
    - zigzag