shubhayansarkar
commited on
Commit
•
a5d30f1
1
Parent(s):
33520ab
Update README.md
Browse files
README.md
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
---
|
2 |
-
license:
|
3 |
library: ultralytics
|
4 |
tags:
|
5 |
- object-detection
|
@@ -32,7 +32,7 @@ import cv
|
|
32 |
|
33 |
# download model
|
34 |
model_path = hf_hub_download(
|
35 |
-
repo_id = "
|
36 |
filename = "model.pt"
|
37 |
)
|
38 |
|
@@ -50,7 +50,6 @@ def inference(image_path):
|
|
50 |
## Training Code
|
51 |
|
52 |
- Dataset Link: [Roboflow Universe](https://universe.roboflow.com/smart2/persondection-61bc2)
|
53 |
-
- Weights & Biases: [Run Details](https://wandb.ai/2wb2ndur/Human-Detection-Thermal-Camera/runs/5j95496q/overview?workspace=user-2wb2ndur)
|
54 |
|
55 |
```python
|
56 |
from ultralytics import YOLO
|
@@ -87,4 +86,4 @@ python: 3.10.13
|
|
87 |
ultralytics: 8.0.206
|
88 |
torch: "2.1.0+cu118"
|
89 |
roboflow: 1.1.9
|
90 |
-
```
|
|
|
1 |
---
|
2 |
+
license: agpl-3.0
|
3 |
library: ultralytics
|
4 |
tags:
|
5 |
- object-detection
|
|
|
32 |
|
33 |
# download model
|
34 |
model_path = hf_hub_download(
|
35 |
+
repo_id = "pitangent-ds/YOLOv8-human-detection-thermal",
|
36 |
filename = "model.pt"
|
37 |
)
|
38 |
|
|
|
50 |
## Training Code
|
51 |
|
52 |
- Dataset Link: [Roboflow Universe](https://universe.roboflow.com/smart2/persondection-61bc2)
|
|
|
53 |
|
54 |
```python
|
55 |
from ultralytics import YOLO
|
|
|
86 |
ultralytics: 8.0.206
|
87 |
torch: "2.1.0+cu118"
|
88 |
roboflow: 1.1.9
|
89 |
+
```
|