Hsueh1001 commited on
Commit
07dbaac
·
1 Parent(s): 8661240

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +11 -4
config.json CHANGED
@@ -1142,10 +1142,17 @@
1142
  "layer_norm_eps": 1e-12,
1143
  "model_type": "yolos",
1144
  "names": [
1145
- "0",
1146
- "1"
 
 
 
 
 
 
 
1147
  ],
1148
- "nc": 2,
1149
  "nms_threshold": 0.1,
1150
  "num_attention_heads": 12,
1151
  "num_channels": 3,
@@ -1157,6 +1164,6 @@
1157
  "torch_dtype": "float32",
1158
  "transformers_version": "4.36.2",
1159
  "use_mid_position_embeddings": true,
1160
- "weights": "10.pt",
1161
  "width_multiple": 1.0
1162
  }
 
1142
  "layer_norm_eps": 1e-12,
1143
  "model_type": "yolos",
1144
  "names": [
1145
+ "person", "bicycle", "car", "motorcycle", "airplane", "bus", "train", "truck", "boat", "traffic light",
1146
+ "fire hydrant", "stop sign", "parking meter", "bench", "bird", "cat", "dog", "horse", "sheep", "cow",
1147
+ "elephant", "bear", "zebra", "giraffe", "backpack", "umbrella", "handbag", "tie", "suitcase", "frisbee",
1148
+ "skis", "snowboard", "sports ball", "kite", "baseball bat", "baseball glove", "skateboard", "surfboard",
1149
+ "tennis racket", "bottle", "wine glass", "cup", "fork", "knife", "spoon", "bowl", "banana", "apple",
1150
+ "sandwich", "orange", "broccoli", "carrot", "hot dog", "pizza", "donut", "cake", "chair", "couch",
1151
+ "potted plant", "bed", "dining table", "toilet", "tv", "laptop", "mouse", "remote", "keyboard", "cell phone",
1152
+ "microwave", "oven", "toaster", "sink", "refrigerator", "book", "clock", "vase", "scissors", "teddy bear",
1153
+ "hair drier", "toothbrush"
1154
  ],
1155
+ "nc": 80,
1156
  "nms_threshold": 0.1,
1157
  "num_attention_heads": 12,
1158
  "num_channels": 3,
 
1164
  "torch_dtype": "float32",
1165
  "transformers_version": "4.36.2",
1166
  "use_mid_position_embeddings": true,
1167
+ "weights": "yolov7.pt",
1168
  "width_multiple": 1.0
1169
  }