okhytrov commited on
Commit
23f50b5
1 Parent(s): 3bb0d2c

Upload DetrForObjectDetection

Browse files
Files changed (2) hide show
  1. config.json +13 -13
  2. model.safetensors +1 -1
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "facebook/detr-resnet-50",
3
  "activation_dropout": 0.0,
4
  "activation_function": "relu",
5
  "architectures": [
@@ -121,18 +121,18 @@
121
  "giou_cost": 2,
122
  "giou_loss_coefficient": 2,
123
  "id2label": {
124
- "0": "LABEL_0",
125
- "1": "LABEL_1",
126
- "2": "LABEL_2",
127
- "3": "LABEL_3",
128
- "4": "LABEL_4",
129
- "5": "LABEL_5",
130
- "6": "LABEL_6",
131
- "7": "LABEL_7",
132
- "8": "LABEL_8",
133
- "9": "LABEL_9",
134
- "10": "LABEL_10",
135
- "11": "LABEL_11"
136
  },
137
  "init_std": 0.02,
138
  "init_xavier_std": 1.0,
 
1
  {
2
+ "_name_or_path": "okhytrov/detr-finetuned-visdrone",
3
  "activation_dropout": 0.0,
4
  "activation_function": "relu",
5
  "architectures": [
 
121
  "giou_cost": 2,
122
  "giou_loss_coefficient": 2,
123
  "id2label": {
124
+ "0": "Ignore",
125
+ "1": "Pedestrian",
126
+ "2": "People",
127
+ "3": "Bicycle",
128
+ "4": "Car",
129
+ "5": "Van",
130
+ "6": "Truck",
131
+ "7": "Tricycle",
132
+ "8": "Awning-tricycle",
133
+ "9": "Bus",
134
+ "10": "Motor",
135
+ "11": "Others"
136
  },
137
  "init_std": 0.02,
138
  "init_xavier_std": 1.0,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c4b00e4923b0c94839e75d061b0f57bf3b3c41e658c9f22b742a28dcb0d632c9
3
  size 166515036
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d4c65bfd4b7269e4df554f5142d3b7c66bd2eb22f745d8cbc8216538dc8c7728
3
  size 166515036