new model

app.py

@@ -4,81 +4,18 @@ import gradio as gr
 
 def classify_image(img):
     from fastai.vision.all import load_learner
-    learn = load_learner('lego-bricks-model.pkl')
+    learn = load_learner('lego-bricks-detecter-model.pkl')
     categories = (
-     #'11214 Bush 3M friction with Cross axle',
-     #'18651 Cross Axle 2M with Snap friction',
-     #'2357 Brick corner 1x2x2',
-     #'3003 Brick 2x2',
-     #'3004 Brick 1x2', 
-     #'3005 Brick 1x1',
-     #'3022 Plate 2x2',
-     #'3023 Plate 1x2',
-     #'3024 Plate 1x1',
-     #'3040 Roof Tile 1x2x45deg',
-     #'3069 Flat Tile 1x2',
-     #'32123 half Bush',
-     #'3673 Peg 2M',
-     #'3713 Bush for Cross Axle',
-     #'3794 Plate 1X2 with 1 Knob',
-     #'6632 Technic Lever 3M', 
-    'Brick_11476',
-    'Brick_12897',
-    'Brick_12899',
-    'Brick_13760',
-    'Brick_16615',
-    'Brick_18896',
-    'Brick_18976',
-    'Brick_21445',
-    'Brick_21980',
-    'Brick_24869',
-    'Brick_2499',
-    'Brick_2610',
-    'Brick_26280',
-    'Brick_28324',
-    'Brick_2921',
-    'Brick_303226',
-    'Brick_30357',
-    'Brick_30407',
-    'Brick_31520',
-    'Brick_32124',
-    'Brick_3479',
-    'Brick_3708',
-    'Brick_4106592',
-    'Brick_41630',
-    'Brick_41748',
-    'Brick_42022',
-    'Brick_42074',
-    'Brick_42604',
-    'Brick_4261453',
-    'Brick_4503',
-    'Brick_4515',
-    'Brick_45407',
-    'Brick_45590',
-    'Brick_48002',
-    'Brick_502',
-    'Brick_50956',
-    'Brick_52216',
-    'Brick_54671',
-    'Brick_602',
-    'Brick_6162',
-    'Brick_6182',
-    'Brick_6191',
-    'Brick_685',
-    'Brick_711',
-    'Brick_81599',
-    'Brick_87995',
-    'Brick_88704',
-    'Brick_92092',
-    'Brick_938',
-    'Brick_94318',
-) 
+        ['Brick_2357', 'Brick_3002', 'Brick_3003', 'Brick_3004', 'Brick_3010', 'Brick_30145', 'Brick_3065', 'Brick_6223', 'Brick_702'],
+        [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 13, 14, 15, 18, 19, 20, 22, 25, 26, 27, 28, 29, 30, 31, 33, 34, 35, 36, 37, 38, 40, 41, 42, 43, 46, 47, 52, 69, 70, 71, 72, 73, 74, 78, 82, 84, 85, 89, 92, 114, 117, 118, 191, 212, 226, 256, 272, 288, 297, 302, 308, 313, 320, 321, 322, 323, 326, 330, 335, 351, 353, 360, 362, 368, 378, 379, 462, 484]
+    )
+
     pred,idx,probs = learn.predict(img)
     return dict(zip(categories, map(float,probs)))
 
-image = gr.inputs.Image(shape=(192, 192))
+image = gr.inputs.Image(shape=(256, 256))
 label = gr.outputs.Label()
-examples = ['0002.png', '201706161606-0002.png', '201706161906-0001.png']
+examples = []
 
 intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
 intf.launch(inline=False)

lego-bricks-detecter-model.pkl

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