sandbox338 commited on
Commit
6e13a93
·
verified ·
1 Parent(s): b0d6592

Upload model_info.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. model_info.json +28 -0
model_info.json ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "ensemble_wildlife_detector",
3
+ "backbone_types": [
4
+ "inception",
5
+ "resnet"
6
+ ],
7
+ "num_classes": 11,
8
+ "class_names": [
9
+ "Antelope",
10
+ "Lion",
11
+ "elephant",
12
+ "zebra",
13
+ "Gorilla",
14
+ "Wolf",
15
+ "Hippopotamus",
16
+ "Leopard",
17
+ "Giraffe",
18
+ "Rhino",
19
+ "Buffalo"
20
+ ],
21
+ "input_size": [
22
+ 800,
23
+ 800
24
+ ],
25
+ "framework": "detectron2",
26
+ "pytorch_version": "2.6.0+cu124",
27
+ "created_by": "wildlife_detection_ensemble"
28
+ }