nragrawal commited on
Commit
d6e2ebe
·
1 Parent(s): 272bcbc

Add model weights

Browse files
Files changed (3) hide show
  1. app.py +20 -10
  2. model/classes.txt +1000 -0
  3. model/model_best.pth +3 -0
app.py CHANGED
@@ -1,18 +1,17 @@
1
  import gradio as gr
2
- from transformers import AutoModelForImageClassification
3
  import torch
4
  import torchvision.transforms as transforms
5
  from PIL import Image
6
  import traceback
7
  import sys
 
8
 
9
- # Load model from Hub instead of local file
10
  def load_model():
11
  try:
12
- model = AutoModelForImageClassification.from_pretrained(
13
- "nragrawal/resnet-imagenet1k",
14
- trust_remote_code=True
15
- )
16
  model.eval()
17
  return model
18
  except Exception as e:
@@ -20,6 +19,15 @@ def load_model():
20
  print(traceback.format_exc())
21
  raise e
22
 
 
 
 
 
 
 
 
 
 
23
  # Preprocessing
24
  transform = transforms.Compose([
25
  transforms.Resize(256),
@@ -29,11 +37,13 @@ transform = transforms.Compose([
29
  std=[0.229, 0.224, 0.225])
30
  ])
31
 
 
 
 
 
32
  # Inference function
33
  def predict(image):
34
  try:
35
- model = load_model()
36
-
37
  # Preprocess image
38
  img = Image.fromarray(image)
39
  img = transform(img).unsqueeze(0)
@@ -45,7 +55,7 @@ def predict(image):
45
 
46
  # Get top 5 predictions
47
  top5_prob, top5_catid = torch.topk(probabilities, 5)
48
- return {f"Class {i}": float(prob) for i, prob in zip(top5_catid, top5_prob)}
49
  except Exception as e:
50
  print(f"Error during prediction: {str(e)}")
51
  print(traceback.format_exc())
@@ -57,7 +67,7 @@ iface = gr.Interface(
57
  inputs=gr.Image(),
58
  outputs=gr.Label(num_top_classes=5),
59
  title="ResNet Image Classification",
60
- description="Upload an image to classify it using ResNet",
61
  allow_flagging="never"
62
  )
63
 
 
1
  import gradio as gr
 
2
  import torch
3
  import torchvision.transforms as transforms
4
  from PIL import Image
5
  import traceback
6
  import sys
7
+ from network import create_model # Import our model architecture
8
 
9
+ # Load model from local checkpoint
10
  def load_model():
11
  try:
12
+ model = create_model(num_classes=1000) # Match your training classes
13
+ checkpoint = torch.load('model/model_best.pth', map_location='cpu')
14
+ model.load_state_dict(checkpoint['model_state_dict'])
 
15
  model.eval()
16
  return model
17
  except Exception as e:
 
19
  print(traceback.format_exc())
20
  raise e
21
 
22
+ # Load ImageNet class labels
23
+ def load_labels():
24
+ try:
25
+ with open('model/classes.txt', 'r') as f:
26
+ labels = [line.strip() for line in f.readlines()]
27
+ return labels
28
+ except:
29
+ return [f"Class_{i}" for i in range(100)] # Fallback to generic labels
30
+
31
  # Preprocessing
32
  transform = transforms.Compose([
33
  transforms.Resize(256),
 
37
  std=[0.229, 0.224, 0.225])
38
  ])
39
 
40
+ # Global variables
41
+ model = load_model()
42
+ labels = load_labels()
43
+
44
  # Inference function
45
  def predict(image):
46
  try:
 
 
47
  # Preprocess image
48
  img = Image.fromarray(image)
49
  img = transform(img).unsqueeze(0)
 
55
 
56
  # Get top 5 predictions
57
  top5_prob, top5_catid = torch.topk(probabilities, 5)
58
+ return {labels[idx]: float(prob) for prob, idx in zip(top5_prob, top5_catid)}
59
  except Exception as e:
60
  print(f"Error during prediction: {str(e)}")
61
  print(traceback.format_exc())
 
67
  inputs=gr.Image(),
68
  outputs=gr.Label(num_top_classes=5),
69
  title="ResNet Image Classification",
70
+ description="Upload an image to classify it using ResNet trained on ImageNet subset",
71
  allow_flagging="never"
72
  )
73
 
model/classes.txt ADDED
@@ -0,0 +1,1000 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ tench
2
+ goldfish
3
+ great_white_shark
4
+ tiger_shark
5
+ hammerhead
6
+ electric_ray
7
+ stingray
8
+ cock
9
+ hen
10
+ ostrich
11
+ brambling
12
+ goldfinch
13
+ house_finch
14
+ junco
15
+ indigo_bunting
16
+ robin
17
+ bulbul
18
+ jay
19
+ magpie
20
+ chickadee
21
+ water_ouzel
22
+ kite
23
+ bald_eagle
24
+ vulture
25
+ great_grey_owl
26
+ European_fire_salamander
27
+ common_newt
28
+ eft
29
+ spotted_salamander
30
+ axolotl
31
+ bullfrog
32
+ tree_frog
33
+ tailed_frog
34
+ loggerhead
35
+ leatherback_turtle
36
+ mud_turtle
37
+ terrapin
38
+ box_turtle
39
+ banded_gecko
40
+ common_iguana
41
+ American_chameleon
42
+ whiptail
43
+ agama
44
+ frilled_lizard
45
+ alligator_lizard
46
+ Gila_monster
47
+ green_lizard
48
+ African_chameleon
49
+ Komodo_dragon
50
+ African_crocodile
51
+ American_alligator
52
+ triceratops
53
+ thunder_snake
54
+ ringneck_snake
55
+ hognose_snake
56
+ green_snake
57
+ king_snake
58
+ garter_snake
59
+ water_snake
60
+ vine_snake
61
+ night_snake
62
+ boa_constrictor
63
+ rock_python
64
+ Indian_cobra
65
+ green_mamba
66
+ sea_snake
67
+ horned_viper
68
+ diamondback
69
+ sidewinder
70
+ trilobite
71
+ harvestman
72
+ scorpion
73
+ black_and_gold_garden_spider
74
+ barn_spider
75
+ garden_spider
76
+ black_widow
77
+ tarantula
78
+ wolf_spider
79
+ tick
80
+ centipede
81
+ black_grouse
82
+ ptarmigan
83
+ ruffed_grouse
84
+ prairie_chicken
85
+ peacock
86
+ quail
87
+ partridge
88
+ African_grey
89
+ macaw
90
+ sulphur-crested_cockatoo
91
+ lorikeet
92
+ coucal
93
+ bee_eater
94
+ hornbill
95
+ hummingbird
96
+ jacamar
97
+ toucan
98
+ drake
99
+ red-breasted_merganser
100
+ goose
101
+ black_swan
102
+ tusker
103
+ echidna
104
+ platypus
105
+ wallaby
106
+ koala
107
+ wombat
108
+ jellyfish
109
+ sea_anemone
110
+ brain_coral
111
+ flatworm
112
+ nematode
113
+ conch
114
+ snail
115
+ slug
116
+ sea_slug
117
+ chiton
118
+ chambered_nautilus
119
+ Dungeness_crab
120
+ rock_crab
121
+ fiddler_crab
122
+ king_crab
123
+ American_lobster
124
+ spiny_lobster
125
+ crayfish
126
+ hermit_crab
127
+ isopod
128
+ white_stork
129
+ black_stork
130
+ spoonbill
131
+ flamingo
132
+ little_blue_heron
133
+ American_egret
134
+ bittern
135
+ crane
136
+ limpkin
137
+ European_gallinule
138
+ American_coot
139
+ bustard
140
+ ruddy_turnstone
141
+ red-backed_sandpiper
142
+ redshank
143
+ dowitcher
144
+ oystercatcher
145
+ pelican
146
+ king_penguin
147
+ albatross
148
+ grey_whale
149
+ killer_whale
150
+ dugong
151
+ sea_lion
152
+ Chihuahua
153
+ Japanese_spaniel
154
+ Maltese_dog
155
+ Pekinese
156
+ Shih-Tzu
157
+ Blenheim_spaniel
158
+ papillon
159
+ toy_terrier
160
+ Rhodesian_ridgeback
161
+ Afghan_hound
162
+ basset
163
+ beagle
164
+ bloodhound
165
+ bluetick
166
+ black-and-tan_coonhound
167
+ Walker_hound
168
+ English_foxhound
169
+ redbone
170
+ borzoi
171
+ Irish_wolfhound
172
+ Italian_greyhound
173
+ whippet
174
+ Ibizan_hound
175
+ Norwegian_elkhound
176
+ otterhound
177
+ Saluki
178
+ Scottish_deerhound
179
+ Weimaraner
180
+ Staffordshire_bullterrier
181
+ American_Staffordshire_terrier
182
+ Bedlington_terrier
183
+ Border_terrier
184
+ Kerry_blue_terrier
185
+ Irish_terrier
186
+ Norfolk_terrier
187
+ Norwich_terrier
188
+ Yorkshire_terrier
189
+ wire-haired_fox_terrier
190
+ Lakeland_terrier
191
+ Sealyham_terrier
192
+ Airedale
193
+ cairn
194
+ Australian_terrier
195
+ Dandie_Dinmont
196
+ Boston_bull
197
+ miniature_schnauzer
198
+ giant_schnauzer
199
+ standard_schnauzer
200
+ Scotch_terrier
201
+ Tibetan_terrier
202
+ silky_terrier
203
+ soft-coated_wheaten_terrier
204
+ West_Highland_white_terrier
205
+ Lhasa
206
+ flat-coated_retriever
207
+ curly-coated_retriever
208
+ golden_retriever
209
+ Labrador_retriever
210
+ Chesapeake_Bay_retriever
211
+ German_short-haired_pointer
212
+ vizsla
213
+ English_setter
214
+ Irish_setter
215
+ Gordon_setter
216
+ Brittany_spaniel
217
+ clumber
218
+ English_springer
219
+ Welsh_springer_spaniel
220
+ cocker_spaniel
221
+ Sussex_spaniel
222
+ Irish_water_spaniel
223
+ kuvasz
224
+ schipperke
225
+ groenendael
226
+ malinois
227
+ briard
228
+ kelpie
229
+ komondor
230
+ Old_English_sheepdog
231
+ Shetland_sheepdog
232
+ collie
233
+ Border_collie
234
+ Bouvier_des_Flandres
235
+ Rottweiler
236
+ German_shepherd
237
+ Doberman
238
+ miniature_pinscher
239
+ Greater_Swiss_Mountain_dog
240
+ Bernese_mountain_dog
241
+ Appenzeller
242
+ EntleBucher
243
+ boxer
244
+ bull_mastiff
245
+ Tibetan_mastiff
246
+ French_bulldog
247
+ Great_Dane
248
+ Saint_Bernard
249
+ Eskimo_dog
250
+ malamute
251
+ Siberian_husky
252
+ dalmatian
253
+ affenpinscher
254
+ basenji
255
+ pug
256
+ Leonberg
257
+ Newfoundland
258
+ Great_Pyrenees
259
+ Samoyed
260
+ Pomeranian
261
+ chow
262
+ keeshond
263
+ Brabancon_griffon
264
+ Pembroke
265
+ Cardigan
266
+ toy_poodle
267
+ miniature_poodle
268
+ standard_poodle
269
+ Mexican_hairless
270
+ timber_wolf
271
+ white_wolf
272
+ red_wolf
273
+ coyote
274
+ dingo
275
+ dhole
276
+ African_hunting_dog
277
+ hyena
278
+ red_fox
279
+ kit_fox
280
+ Arctic_fox
281
+ grey_fox
282
+ tabby
283
+ tiger_cat
284
+ Persian_cat
285
+ Siamese_cat
286
+ Egyptian_cat
287
+ cougar
288
+ lynx
289
+ leopard
290
+ snow_leopard
291
+ jaguar
292
+ lion
293
+ tiger
294
+ cheetah
295
+ brown_bear
296
+ American_black_bear
297
+ ice_bear
298
+ sloth_bear
299
+ mongoose
300
+ meerkat
301
+ tiger_beetle
302
+ ladybug
303
+ ground_beetle
304
+ long-horned_beetle
305
+ leaf_beetle
306
+ dung_beetle
307
+ rhinoceros_beetle
308
+ weevil
309
+ fly
310
+ bee
311
+ ant
312
+ grasshopper
313
+ cricket
314
+ walking_stick
315
+ cockroach
316
+ mantis
317
+ cicada
318
+ leafhopper
319
+ lacewing
320
+ dragonfly
321
+ damselfly
322
+ admiral
323
+ ringlet
324
+ monarch
325
+ cabbage_butterfly
326
+ sulphur_butterfly
327
+ lycaenid
328
+ starfish
329
+ sea_urchin
330
+ sea_cucumber
331
+ wood_rabbit
332
+ hare
333
+ Angora
334
+ hamster
335
+ porcupine
336
+ fox_squirrel
337
+ marmot
338
+ beaver
339
+ guinea_pig
340
+ sorrel
341
+ zebra
342
+ hog
343
+ wild_boar
344
+ warthog
345
+ hippopotamus
346
+ ox
347
+ water_buffalo
348
+ bison
349
+ ram
350
+ bighorn
351
+ ibex
352
+ hartebeest
353
+ impala
354
+ gazelle
355
+ Arabian_camel
356
+ llama
357
+ weasel
358
+ mink
359
+ polecat
360
+ black-footed_ferret
361
+ otter
362
+ skunk
363
+ badger
364
+ armadillo
365
+ three-toed_sloth
366
+ orangutan
367
+ gorilla
368
+ chimpanzee
369
+ gibbon
370
+ siamang
371
+ guenon
372
+ patas
373
+ baboon
374
+ macaque
375
+ langur
376
+ colobus
377
+ proboscis_monkey
378
+ marmoset
379
+ capuchin
380
+ howler_monkey
381
+ titi
382
+ spider_monkey
383
+ squirrel_monkey
384
+ Madagascar_cat
385
+ indri
386
+ Indian_elephant
387
+ African_elephant
388
+ lesser_panda
389
+ giant_panda
390
+ barracouta
391
+ eel
392
+ coho
393
+ rock_beauty
394
+ anemone_fish
395
+ sturgeon
396
+ gar
397
+ lionfish
398
+ puffer
399
+ abacus
400
+ abaya
401
+ academic_gown
402
+ accordion
403
+ acoustic_guitar
404
+ aircraft_carrier
405
+ airliner
406
+ airship
407
+ altar
408
+ ambulance
409
+ amphibian
410
+ analog_clock
411
+ apiary
412
+ apron
413
+ ashcan
414
+ assault_rifle
415
+ backpack
416
+ bakery
417
+ balance_beam
418
+ balloon
419
+ ballpoint
420
+ Band_Aid
421
+ banjo
422
+ bannister
423
+ barbell
424
+ barber_chair
425
+ barbershop
426
+ barn
427
+ barometer
428
+ barrel
429
+ barrow
430
+ baseball
431
+ basketball
432
+ bassinet
433
+ bassoon
434
+ bathing_cap
435
+ bath_towel
436
+ bathtub
437
+ beach_wagon
438
+ beacon
439
+ beaker
440
+ bearskin
441
+ beer_bottle
442
+ beer_glass
443
+ bell_cote
444
+ bib
445
+ bicycle-built-for-two
446
+ bikini
447
+ binder
448
+ binoculars
449
+ birdhouse
450
+ boathouse
451
+ bobsled
452
+ bolo_tie
453
+ bonnet
454
+ bookcase
455
+ bookshop
456
+ bottlecap
457
+ bow
458
+ bow_tie
459
+ brass
460
+ brassiere
461
+ breakwater
462
+ breastplate
463
+ broom
464
+ bucket
465
+ buckle
466
+ bulletproof_vest
467
+ bullet_train
468
+ butcher_shop
469
+ cab
470
+ caldron
471
+ candle
472
+ cannon
473
+ canoe
474
+ can_opener
475
+ cardigan
476
+ car_mirror
477
+ carousel
478
+ carpenters_kit
479
+ carton
480
+ car_wheel
481
+ cash_machine
482
+ cassette
483
+ cassette_player
484
+ castle
485
+ catamaran
486
+ CD_player
487
+ cello
488
+ cellular_telephone
489
+ chain
490
+ chainlink_fence
491
+ chain_mail
492
+ chain_saw
493
+ chest
494
+ chiffonier
495
+ chime
496
+ china_cabinet
497
+ Christmas_stocking
498
+ church
499
+ cinema
500
+ cleaver
501
+ cliff_dwelling
502
+ cloak
503
+ clog
504
+ cocktail_shaker
505
+ coffee_mug
506
+ coffeepot
507
+ coil
508
+ combination_lock
509
+ computer_keyboard
510
+ confectionery
511
+ container_ship
512
+ convertible
513
+ corkscrew
514
+ cornet
515
+ cowboy_boot
516
+ cowboy_hat
517
+ cradle
518
+ crane
519
+ crash_helmet
520
+ crate
521
+ crib
522
+ Crock_Pot
523
+ croquet_ball
524
+ crutch
525
+ cuirass
526
+ dam
527
+ desk
528
+ desktop_computer
529
+ dial_telephone
530
+ diaper
531
+ digital_clock
532
+ digital_watch
533
+ dining_table
534
+ dishrag
535
+ dishwasher
536
+ disk_brake
537
+ dock
538
+ dogsled
539
+ dome
540
+ doormat
541
+ drilling_platform
542
+ drum
543
+ drumstick
544
+ dumbbell
545
+ Dutch_oven
546
+ electric_fan
547
+ electric_guitar
548
+ electric_locomotive
549
+ entertainment_center
550
+ envelope
551
+ espresso_maker
552
+ face_powder
553
+ feather_boa
554
+ file
555
+ fireboat
556
+ fire_engine
557
+ fire_screen
558
+ flagpole
559
+ flute
560
+ folding_chair
561
+ football_helmet
562
+ forklift
563
+ fountain
564
+ fountain_pen
565
+ four-poster
566
+ freight_car
567
+ French_horn
568
+ frying_pan
569
+ fur_coat
570
+ garbage_truck
571
+ gasmask
572
+ gas_pump
573
+ goblet
574
+ go-kart
575
+ golf_ball
576
+ golfcart
577
+ gondola
578
+ gong
579
+ gown
580
+ grand_piano
581
+ greenhouse
582
+ grille
583
+ grocery_store
584
+ guillotine
585
+ hair_slide
586
+ hair_spray
587
+ half_track
588
+ hammer
589
+ hamper
590
+ hand_blower
591
+ hand-held_computer
592
+ handkerchief
593
+ hard_disc
594
+ harmonica
595
+ harp
596
+ harvester
597
+ hatchet
598
+ holster
599
+ home_theater
600
+ honeycomb
601
+ hook
602
+ hoopskirt
603
+ horizontal_bar
604
+ horse_cart
605
+ hourglass
606
+ iPod
607
+ iron
608
+ jack-o-lantern
609
+ jean
610
+ jeep
611
+ jersey
612
+ jigsaw_puzzle
613
+ jinrikisha
614
+ joystick
615
+ kimono
616
+ knee_pad
617
+ knot
618
+ lab_coat
619
+ ladle
620
+ lampshade
621
+ laptop
622
+ lawn_mower
623
+ lens_cap
624
+ letter_opener
625
+ library
626
+ lifeboat
627
+ lighter
628
+ limousine
629
+ liner
630
+ lipstick
631
+ Loafer
632
+ lotion
633
+ loudspeaker
634
+ loupe
635
+ lumbermill
636
+ magnetic_compass
637
+ mailbag
638
+ mailbox
639
+ maillot
640
+ maillot
641
+ manhole_cover
642
+ maraca
643
+ marimba
644
+ mask
645
+ matchstick
646
+ maypole
647
+ maze
648
+ measuring_cup
649
+ medicine_chest
650
+ megalith
651
+ microphone
652
+ microwave
653
+ military_uniform
654
+ milk_can
655
+ minibus
656
+ miniskirt
657
+ minivan
658
+ missile
659
+ mitten
660
+ mixing_bowl
661
+ mobile_home
662
+ Model_T
663
+ modem
664
+ monastery
665
+ monitor
666
+ moped
667
+ mortar
668
+ mortarboard
669
+ mosque
670
+ mosquito_net
671
+ motor_scooter
672
+ mountain_bike
673
+ mountain_tent
674
+ mouse
675
+ mousetrap
676
+ moving_van
677
+ muzzle
678
+ nail
679
+ neck_brace
680
+ necklace
681
+ nipple
682
+ notebook
683
+ obelisk
684
+ oboe
685
+ ocarina
686
+ odometer
687
+ oil_filter
688
+ organ
689
+ oscilloscope
690
+ overskirt
691
+ oxcart
692
+ oxygen_mask
693
+ packet
694
+ paddle
695
+ paddlewheel
696
+ padlock
697
+ paintbrush
698
+ pajama
699
+ palace
700
+ panpipe
701
+ paper_towel
702
+ parachute
703
+ parallel_bars
704
+ park_bench
705
+ parking_meter
706
+ passenger_car
707
+ patio
708
+ pay-phone
709
+ pedestal
710
+ pencil_box
711
+ pencil_sharpener
712
+ perfume
713
+ Petri_dish
714
+ photocopier
715
+ pick
716
+ pickelhaube
717
+ picket_fence
718
+ pickup
719
+ pier
720
+ piggy_bank
721
+ pill_bottle
722
+ pillow
723
+ ping-pong_ball
724
+ pinwheel
725
+ pirate
726
+ pitcher
727
+ plane
728
+ planetarium
729
+ plastic_bag
730
+ plate_rack
731
+ plow
732
+ plunger
733
+ Polaroid_camera
734
+ pole
735
+ police_van
736
+ poncho
737
+ pool_table
738
+ pop_bottle
739
+ pot
740
+ potters_wheel
741
+ power_drill
742
+ prayer_rug
743
+ printer
744
+ prison
745
+ projectile
746
+ projector
747
+ puck
748
+ punching_bag
749
+ purse
750
+ quill
751
+ quilt
752
+ racer
753
+ racket
754
+ radiator
755
+ radio
756
+ radio_telescope
757
+ rain_barrel
758
+ recreational_vehicle
759
+ reel
760
+ reflex_camera
761
+ refrigerator
762
+ remote_control
763
+ restaurant
764
+ revolver
765
+ rifle
766
+ rocking_chair
767
+ rotisserie
768
+ rubber_eraser
769
+ rugby_ball
770
+ rule
771
+ running_shoe
772
+ safe
773
+ safety_pin
774
+ saltshaker
775
+ sandal
776
+ sarong
777
+ sax
778
+ scabbard
779
+ scale
780
+ school_bus
781
+ schooner
782
+ scoreboard
783
+ screen
784
+ screw
785
+ screwdriver
786
+ seat_belt
787
+ sewing_machine
788
+ shield
789
+ shoe_shop
790
+ shoji
791
+ shopping_basket
792
+ shopping_cart
793
+ shovel
794
+ shower_cap
795
+ shower_curtain
796
+ ski
797
+ ski_mask
798
+ sleeping_bag
799
+ slide_rule
800
+ sliding_door
801
+ slot
802
+ snorkel
803
+ snowmobile
804
+ snowplow
805
+ soap_dispenser
806
+ soccer_ball
807
+ sock
808
+ solar_dish
809
+ sombrero
810
+ soup_bowl
811
+ space_bar
812
+ space_heater
813
+ space_shuttle
814
+ spatula
815
+ speedboat
816
+ spider_web
817
+ spindle
818
+ sports_car
819
+ spotlight
820
+ stage
821
+ steam_locomotive
822
+ steel_arch_bridge
823
+ steel_drum
824
+ stethoscope
825
+ stole
826
+ stone_wall
827
+ stopwatch
828
+ stove
829
+ strainer
830
+ streetcar
831
+ stretcher
832
+ studio_couch
833
+ stupa
834
+ submarine
835
+ suit
836
+ sundial
837
+ sunglass
838
+ sunglasses
839
+ sunscreen
840
+ suspension_bridge
841
+ swab
842
+ sweatshirt
843
+ swimming_trunks
844
+ swing
845
+ switch
846
+ syringe
847
+ table_lamp
848
+ tank
849
+ tape_player
850
+ teapot
851
+ teddy
852
+ television
853
+ tennis_ball
854
+ thatch
855
+ theater_curtain
856
+ thimble
857
+ thresher
858
+ throne
859
+ tile_roof
860
+ toaster
861
+ tobacco_shop
862
+ toilet_seat
863
+ torch
864
+ totem_pole
865
+ tow_truck
866
+ toyshop
867
+ tractor
868
+ trailer_truck
869
+ tray
870
+ trench_coat
871
+ tricycle
872
+ trimaran
873
+ tripod
874
+ triumphal_arch
875
+ trolleybus
876
+ trombone
877
+ tub
878
+ turnstile
879
+ typewriter_keyboard
880
+ umbrella
881
+ unicycle
882
+ upright
883
+ vacuum
884
+ vase
885
+ vault
886
+ velvet
887
+ vending_machine
888
+ vestment
889
+ viaduct
890
+ violin
891
+ volleyball
892
+ waffle_iron
893
+ wall_clock
894
+ wallet
895
+ wardrobe
896
+ warplane
897
+ washbasin
898
+ washer
899
+ water_bottle
900
+ water_jug
901
+ water_tower
902
+ whiskey_jug
903
+ whistle
904
+ wig
905
+ window_screen
906
+ window_shade
907
+ Windsor_tie
908
+ wine_bottle
909
+ wing
910
+ wok
911
+ wooden_spoon
912
+ wool
913
+ worm_fence
914
+ wreck
915
+ yawl
916
+ yurt
917
+ web_site
918
+ comic_book
919
+ crossword_puzzle
920
+ street_sign
921
+ traffic_light
922
+ book_jacket
923
+ menu
924
+ plate
925
+ guacamole
926
+ consomme
927
+ hot_pot
928
+ trifle
929
+ ice_cream
930
+ ice_lolly
931
+ French_loaf
932
+ bagel
933
+ pretzel
934
+ cheeseburger
935
+ hotdog
936
+ mashed_potato
937
+ head_cabbage
938
+ broccoli
939
+ cauliflower
940
+ zucchini
941
+ spaghetti_squash
942
+ acorn_squash
943
+ butternut_squash
944
+ cucumber
945
+ artichoke
946
+ bell_pepper
947
+ cardoon
948
+ mushroom
949
+ Granny_Smith
950
+ strawberry
951
+ orange
952
+ lemon
953
+ fig
954
+ pineapple
955
+ banana
956
+ jackfruit
957
+ custard_apple
958
+ pomegranate
959
+ hay
960
+ carbonara
961
+ chocolate_sauce
962
+ dough
963
+ meat_loaf
964
+ pizza
965
+ potpie
966
+ burrito
967
+ red_wine
968
+ espresso
969
+ cup
970
+ eggnog
971
+ alp
972
+ bubble
973
+ cliff
974
+ coral_reef
975
+ geyser
976
+ lakeside
977
+ promontory
978
+ sandbar
979
+ seashore
980
+ valley
981
+ volcano
982
+ ballplayer
983
+ groom
984
+ scuba_diver
985
+ rapeseed
986
+ daisy
987
+ yellow_ladys_slipper
988
+ corn
989
+ acorn
990
+ hip
991
+ buckeye
992
+ coral_fungus
993
+ agaric
994
+ gyromitra
995
+ stinkhorn
996
+ earthstar
997
+ hen-of-the-woods
998
+ bolete
999
+ ear
1000
+ toilet_tissue
model/model_best.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8bb6d6d8a6c22c29e5d22ee98ac12fe8e7c61365972feb61a0b78c368ab5f07e
3
+ size 204828790