update model
Browse files- config.json +17 -0
- last_state_dict.pth +3 -0
- 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}
|