Abdu07 commited on
Commit
2192c72
·
verified ·
1 Parent(s): 4e49384

update model

Browse files
Files changed (3) hide show
  1. config.json +17 -0
  2. last_state_dict.pth +3 -0
  3. obj_label_mapping.json +1 -0
config.json ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "backbone": "resnet50",
3
+ "num_obj_classes": 100,
4
+ "img_size": 224,
5
+ "normalize_mean": [
6
+ 0.485,
7
+ 0.456,
8
+ 0.406
9
+ ],
10
+ "normalize_std": [
11
+ 0.229,
12
+ 0.224,
13
+ 0.225
14
+ ],
15
+ "bin_head": "bce_logits_1d",
16
+ "threshold_gen": 0.5
17
+ }
last_state_dict.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7639c6e72a638bbf19dcc5371af7a3aba14ddfddfa170244a645510fc3b0dd47
3
+ size 95188126
obj_label_mapping.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"apple": 0, "aquarium_fish": 1, "baby": 2, "bear": 3, "beaver": 4, "bed": 5, "bee": 6, "beetle": 7, "bicycle": 8, "bottle": 9, "bowl": 10, "boy": 11, "bridge": 12, "bus": 13, "butterfly": 14, "camel": 15, "can": 16, "castle": 17, "caterpillar": 18, "cattle": 19, "chair": 20, "chimpanzee": 21, "clock": 22, "cloud": 23, "cockroach": 24, "couch": 25, "cra": 26, "crocodile": 27, "cup": 28, "dinosaur": 29, "dolphin": 30, "elephant": 31, "flatfish": 32, "forest": 33, "fox": 34, "girl": 35, "hamster": 36, "house": 37, "kangaroo": 38, "keyboard": 39, "lamp": 40, "lawn_mower": 41, "leopard": 42, "lion": 43, "lizard": 44, "lobster": 45, "man": 46, "maple_tree": 47, "motorcycle": 48, "mountain": 49, "mouse": 50, "mushroom": 51, "oak_tree": 52, "orange": 53, "orchid": 54, "otter": 55, "palm_tree": 56, "pear": 57, "pickup_truck": 58, "pine_tree": 59, "plain": 60, "plate": 61, "poppy": 62, "porcupine": 63, "possum": 64, "rabbit": 65, "raccoon": 66, "ray": 67, "road": 68, "rocket": 69, "rose": 70, "sea": 71, "seal": 72, "shark": 73, "shrew": 74, "skunk": 75, "skyscraper": 76, "snail": 77, "snake": 78, "spider": 79, "squirrel": 80, "streetcar": 81, "sunflower": 82, "sweet_pepper": 83, "table": 84, "tank": 85, "telephone": 86, "television": 87, "tiger": 88, "tractor": 89, "train": 90, "trout": 91, "tulip": 92, "turtle": 93, "wardrobe": 94, "whale": 95, "willow_tree": 96, "wolf": 97, "woman": 98, "worm": 99}