morancium commited on
Commit
ee202e7
·
verified ·
1 Parent(s): fe53a1f

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +26 -11
config.json CHANGED
@@ -1,14 +1,29 @@
1
  {
2
- "task": "detect",
3
- "data": "/content/Solar-Rooftops-1/data.yaml",
4
- "imgsz": 640,
5
- "single_cls": false,
6
- "model": "YOLOv8n_roof.pt",
7
- "id2label": {
8
- "0": "roof"
 
 
 
 
 
 
 
 
 
 
 
9
  },
10
- "label2id": {
11
- "roof": 0
12
- },
13
- "model_type": "yolos"
 
 
 
 
14
  }
 
1
  {
2
+ "model_name": "DetectionModel",
3
+ "model_parameters": {
4
+ "num_classes": 1,
5
+ "image_size": 640,
6
+ "architecture": "unknown",
7
+ "anchors": "default",
8
+ "depth_multiple": 0.33,
9
+ "width_multiple": 0.5,
10
+ "stride": [
11
+ 8.0,
12
+ 16.0,
13
+ 32.0
14
+ ]
15
+ },
16
+ "training_parameters": {
17
+ "batch_size": "default",
18
+ "learning_rate": "default",
19
+ "epochs": "default"
20
  },
21
+ "model_type": "yolos",
22
+ "id2label": {
23
+ "0": "roof"
24
+ },
25
+ "label2id": {
26
+ "roof": 0
27
+ }
28
+
29
  }