arnabdhar commited on
Commit
c2c61aa
1 Parent(s): dc03a13

Added model.pt

Browse files
README.md CHANGED
@@ -1,3 +1,63 @@
1
  ---
2
  license: mit
 
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: mit
3
+ library: ultralytics
4
+ tags:
5
+ - object-detection
6
+ - pytorch
7
+ - ultralytics
8
+ - roboflow-universe
9
+ - human-detection
10
+ - yolov8
11
  ---
12
+ # Human Detection using Thermal Camera
13
+
14
+ ## Use Case
15
+
16
+ ## Evaluation Metrics
17
+
18
+ __Weights & Biases__: [Run Details](https://wandb.ai/2wb2ndur/Human-Detection-Thermal-Camera/runs/5j95496q/overview?workspace=user-2wb2ndur)
19
+
20
+ ### Training Graphs
21
+
22
+ ![Training Graphs](./training_artifacts/results.png)
23
+
24
+ ## Training Code
25
+
26
+ - Dataset Link: [Roboflow Universe](https://universe.roboflow.com/smart2/persondection-61bc2)
27
+
28
+ ```python
29
+ from ultralytics import YOLO
30
+ import torch
31
+
32
+ # load model
33
+ model = YOLO("yolov8n.pt")
34
+
35
+ # hyper parameters
36
+ hyperparams = {
37
+ "batch": 32,
38
+ "epochs": 30,
39
+ "imgsz": [640, 480],
40
+ "optimizer": "AdamW",
41
+ "cos_lr": True,
42
+ "lr0": 3e-5,
43
+ "warmup_epochs": 10
44
+ }
45
+
46
+ # start training
47
+ model.train(
48
+ device = 'cuda' if torch.cuda.is_available() else 'cpu',
49
+ data = "data.yaml",
50
+ **hyperparams
51
+ )
52
+ ```
53
+
54
+ - Click here for: [Training Arguments](./training_artifacts/args.yaml)
55
+
56
+ ## Libraries
57
+
58
+ ```yaml
59
+ python: 3.10.13
60
+ ultralytics: 8.0.206
61
+ torch: "2.1.0+cu118"
62
+ roboflow: 1.1.9
63
+ ```
events.out.tfevents.1699156391.66327ba9d198.4870.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4b0211290e4600ad85a77be909aa628de21b6612245d33b5c85cc861f52cb0db
3
+ size 764723
model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:63181e6c024a8a218020932b516bfb9547130e03e07c8d4671cb2973df038bef
3
+ size 6234201
training_artifacts/args.yaml ADDED
@@ -0,0 +1,100 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ task: detect
2
+ mode: train
3
+ model: yolov8n.pt
4
+ data: data.yaml
5
+ epochs: 30
6
+ patience: 50
7
+ batch: 32
8
+ imgsz:
9
+ - 640
10
+ - 480
11
+ save: true
12
+ save_period: -1
13
+ cache: false
14
+ device: 0
15
+ workers: 8
16
+ project: Human-Detection-Thermal-Camera
17
+ name: radiant-armadillo-1
18
+ exist_ok: false
19
+ pretrained: true
20
+ optimizer: AdamW
21
+ verbose: true
22
+ seed: 0
23
+ deterministic: true
24
+ single_cls: false
25
+ rect: false
26
+ cos_lr: true
27
+ close_mosaic: 10
28
+ resume: false
29
+ amp: true
30
+ fraction: 1.0
31
+ profile: false
32
+ freeze: null
33
+ overlap_mask: true
34
+ mask_ratio: 4
35
+ dropout: 0.0
36
+ val: true
37
+ split: val
38
+ save_json: false
39
+ save_hybrid: false
40
+ conf: null
41
+ iou: 0.7
42
+ max_det: 300
43
+ half: false
44
+ dnn: false
45
+ plots: true
46
+ source: null
47
+ show: false
48
+ save_txt: false
49
+ save_conf: false
50
+ save_crop: false
51
+ show_labels: true
52
+ show_conf: true
53
+ vid_stride: 1
54
+ stream_buffer: false
55
+ line_width: null
56
+ visualize: false
57
+ augment: false
58
+ agnostic_nms: false
59
+ classes: null
60
+ retina_masks: false
61
+ boxes: true
62
+ format: torchscript
63
+ keras: false
64
+ optimize: false
65
+ int8: false
66
+ dynamic: false
67
+ simplify: false
68
+ opset: null
69
+ workspace: 4
70
+ nms: false
71
+ lr0: 3.0e-05
72
+ lrf: 0.01
73
+ momentum: 0.937
74
+ weight_decay: 0.0005
75
+ warmup_epochs: 10
76
+ warmup_momentum: 0.8
77
+ warmup_bias_lr: 0.1
78
+ box: 7.5
79
+ cls: 0.5
80
+ dfl: 1.5
81
+ pose: 12.0
82
+ kobj: 1.0
83
+ label_smoothing: 0.0
84
+ nbs: 64
85
+ hsv_h: 0.015
86
+ hsv_s: 0.7
87
+ hsv_v: 0.4
88
+ degrees: 0.0
89
+ translate: 0.1
90
+ scale: 0.5
91
+ shear: 0.0
92
+ perspective: 0.0
93
+ flipud: 0.0
94
+ fliplr: 0.5
95
+ mosaic: 1.0
96
+ mixup: 0.0
97
+ copy_paste: 0.0
98
+ cfg: null
99
+ tracker: botsort.yaml
100
+ save_dir: Human-Detection-Thermal-Camera/radiant-armadillo-1
training_artifacts/results.png ADDED